时间戳

首页 > 解决方案 > 熊猫：过滤大日期框并在两个不同的目录中写入 to_csv？

python - 熊猫：过滤大日期框并在两个不同的目录中写入 to_csv？

问题描述

dir_1 = "C:\\Users\\kam\\"
dir_2 = "C:\\Users\\sam\\"

data = {'Name' : ['tom','bul','zack','doll','viru','sam','ram','zori'],'country':['CA','missing','NZ','IN','CA','missing', 'NZ', 'missing']}
df = pd.DataFrame(data)
print(df.to_string(index=False))

Name  country
tom       CA
bul  missing
zack       NZ
doll       IN
viru      CA
sam  missing
ram      NZ
zori  missing


[filename[1].to_csv(dir_1 + f'{filename[0]}.csv', index=False) for filename in df.groupby('country', axis='index')]

在上面的代码中，我试图根据“国家”将单独的 csv 文件保存在“dir_1”中，并将 csv 文件名分配为国家名称（CA.csv、NZD.csv、IN.csv、missing.csv）。但现在我想将 dir_2 中的“缺失”国家/地区保存为（missing.csv），其余保存在 dir_1（CA.csv、NZD.csv、IN.csv）中。我们如何扩展上面的代码来解决这个目的？

标签： pythonpandas

解决方案

[filename[1].to_csv(dir_1 + f'{filename[0]}.csv', index=False) if filename[0]!='' else filename[1].to_csv(dir_2 + 'missing.csv', index=False) for filename in df.groupby('country', axis='index')]

推荐阅读