在 Dataframe 中按列分组并为所有组创建单独的 csv

Question

我有一个 100kb 的巨大 CSV 文件，其中包含记录。如下示例

city         employee

california   jhon
delhi        kumar
us           raj
california   brakers
us           kroja
... 

所以我想将它们分组并存储在单独的 CSV 文件中

我上面例子的输出结果是

city         employee

california   jhon
california   brakers
delhi        kumar
us           raj
us           kroja

我是否必须在数据框中按城市排序/分组，然后有一个 for 循环并遍历整个数据框？

她是我的密码

New_data=pd.read_csv("citynames.csv",encoding = "ISO-8859-1") 
Aftergrouping_data=New_data.groupby("city")
#print(Aftergrouping_data.groups)

for name,group in Aftergrouping_data:
    print(name)
    group.to_csv("aftergrouping.csv".format(name), index=False)

问题在于它覆盖了 for 循环中的值并显示最后一组数据。但我想要文件中的所有分组数据。

提前谢谢

Answer 1

从你的描述来看，两个方面似乎很清楚：

1. 您想要排序数据而不是分组数据。您可以在没有 groupby 对象的情况下对数据进行排序。
2. 您希望将数据框导出到单个 csv 文件。此任务无需使用迭代。

您可以简单地使用sort_values 后跟to_csv：

# read file
df = pd.read_csv('citynames.csv', encoding='ISO-8859-1')

# sort by 'city', ascending
df = df.sort_values('city')

# export to csv
df.to_csv('citynames_out.csv', index=False)