如何使用loop命令切片,创建新df并使用特定名称保存?

时间:2019-07-16 22:44:27

标签: python pandas

我的数据框是:

[Columns Displayed][1]
(3974 rows × 8 columns)

如果我输入df["Country"].value_counts(),则输出结果将是10到46之间的值。我只想对value_counts()结果中具有46个国家的子集进行分组,将每个国家分别保存为新的df使用“国家/地区代码”命名df,并使用“国家/地区代码”将其保存为csv文件名的一部分。

例如,我用乌拉圭(国家代码:URY)手动做过:

painel_URY = painel3[painel3.Country =="Uruguay"]

然后,我保存了它:

painel_URY.to_csv(r"C:\pasta1\pasta2\pasta3\painel_URY.csv")

我尝试了什么? 我试图为每个和join()创建一个循环,但我失败了。我想知道是否有可能。

1 个答案:

答案 0 :(得分:0)

使用groupby获取每个国家/地区的数据框,并保存46行的数据框

for country, group in df.groupby('country'):
    if len(group) == 46:
        group.to_csv(f'./{country}.csv')