首页 > 解决方案 > 遍历字典并将字典中的每个数据帧保存为csv?

问题描述

我有一个如下所示的数据框,但包含数千行数据

ID#  Segment Sub-Segment Account Manager Name
1234 America Midwest     Walmart Bobby Jindal
4567 America Midwest     Target  Bobby Jindal
5678 America Northeast   Dunkin  Peter Thiel

我的最终目标是为 Manager Name 列中的每个唯一值创建一个单独的数据框,并将数据框标记为 Manager 的名称。所以上面的数据集会输出两个dfs

所以我决定创建一个字典,现在已经创建了唯一的数据帧,但我现在遇到的问题是我想将每个数据帧保存为 csv,命名约定是经理名称,所以“Bobby Jindal.csv”

df_sliced={}
for name in df['Manager Name'].unique():
    df_sliced[name]=df[df['Manager Name']==name]

我不确定下一步是将dict中的每个数据帧保存为csv。你能帮忙解决一下语法吗?

标签: pythonpandasdictionary

解决方案


您可以使用以下内容进一步简化代码:

for name, g in df.groupby('Manager Name'):
    g.to_csv(f'{name}.csv')

推荐阅读