如何使用for循环将Pandas数据帧的子集另存为单独的csv文件?

时间:2018-12-28 19:13:20

标签: python pandas

我有一个csv文件,其中包含一本杂志审阅的葡萄酒信息。到目前为止,我已经将该文件作为Pandas数据框导入,并制作了该数据框的子集,其中仅包含雷司令葡萄酒的条目。然后,我列出了此子集中“国家/地区”列的所有唯一值的列表。我尝试遍历该列表,为每个国家/地区创建一个新的数据框,然后使用以下代码将其另存为新的csv文件:

for country in c :
...:     cond = df1['country'] == country
...:     df2 = df1[cond]
...:     df2.to_csv(country)

其中c是国家/地区列表。我确实收到了令人讨厌的错误消息,但是文件确实显示在我的工作目录中,但没有.csv扩展名。看起来只是纯文本文件。我想保存文件并以“ country.csv”之类的国家命名。我知道他的文件名的to_csv参数需要是一个字符串,但是我不知道如何为每次迭代(每个新的csv文件)更改它。如果我可以做一些类似的事情来为每个国家/地区制作单独的数据框并命名为类似df_country的名称,而不用一一制作的话,那也很好。我该怎么办?

0 个答案:

没有答案