from pathlib import Path
import pandas as pd
p = Path('csv_files/Data/') # path to files
files = list(p.glob('*.csv')) # list of files
df_dict = dict()
for file in files:
df = pd.read_csv(file) # create dataframe from file
df.dropna(subset=['A', 'B'], inplace=True)
df['C'] = df['A'] / df['B']
key = f'filtered_{file.stem}'
df_dict[key] = df[(df.C >= 50)] # create a new dataframe and store it in a dictionary
# get list of keys
print(df_dict.keys())
# access a dataframe
df_dict['filtered_file_name']
# save dataframes
for k, v in df_dict.items():
v.to_csv(f'{k}.csv')
pathlib
查找所有文件.stem
获取文件名,该文件名可用作存储的数据帧的key
的一部分f串
路径库
相关问题 更多 >
编程相关推荐