如何迭代定义数据框并将其导出为csv

时间:2019-09-26 21:48:40

标签: python pandas

我要选择包含星期几的行,并按1:1进行保存。

Monday_df = df.loc[df['day_of_week'] == 'Monday']
Tuesday_df = df.loc[df['day_of_week'] == 'Tuesday']
...
Sunday_df = df.loc[df['day_of_week'] == 'Sunday']

然后按1导出1:

Monday_df.to_csv('Monday.csv',index=False)
Tuesday_df.to_csv('Tuesday.csv',index=False)
...
Sunday_df.to_csv('Sunday.csv',index=False)

问题:

我可以分两个步骤执行这14个步骤(定义days(7),然后导出days(7))吗?

1 个答案:

答案 0 :(得分:2)

for day in ['Monday', 'Tuesday', 'Wednesday', 'Thursday', 'Friday', 'Saturday', 'Sunday']:
    day_df = df.loc[df['day_of_week'] == day]
    day_df.to_csv(f'{day}.csv', index=False)

这将循环显示星期几的列表。对于每个特定日期,它都会创建该特定日期的子集数据框,然后将其写出到文件中。

我在这里使用f-strings动态创建输出csv名称。因此,该解决方案至少需要python 3.6才能运行。您可以使用.format或带有百分号而不是百分号的超老烦人格式。