python - 将 pandas 组保存到单独的 CSV 文件

Question

是否可以在不使用 .to_csv 的情况下将多个 df 返回到独立的 csv。使用下面的代码，我手动返回所需的值并将其导出到 csv。如果我只有几个文件要导出，那么这是可以的，但如果有大量文件或文件名不断更改数据集，则可能会很麻烦。

如果您有特定值列表并将其导出到 csv，是否有更有效的方法来返回所需值？

import pandas as pd

d = ({
    'C' : ['08:00:00','XX','08:10:00','XX','08:41:42','XX','08:50:00','XX', '09:00:00', 'XX','09:15:00','XX','09:21:00','XX','09:30:00','XX','09:40:00','XX'],
    'D' : ['Home','Home','Home','Home','Away','Away','Shops','Shops','Away','Away','Shops','Shops','Home','Home','Away','Away','Home','Home'],
    'E' : ['Num:','','Num:','','Num:','','Num:','','Num:', '','Num:','','Num:','','Num:', '','Num:', ''],
    'F' : ['1','','1','','1','','1','','1', '','2','','2','','1', '','2',''],   
    'A' : ['A','','A','','A','','A','','A','','A','','A','','A','','A',''],           
    'B' : ['Stop','','Res','','Stop','','Start','','Res','','Stop','','Res','','Start','','Start','']
    })

df = pd.DataFrame(data=d)

#List of designated places
values = ['Home', 'Away', 'Shops']

#Export to csv
Home = df.loc[df['D'] == 'Home'].to_csv('Home.csv')
Away = df.loc[df['D'] == 'Away'].to_csv('Away.csv')
Shops = df.loc[df['D'] == 'Shops'].to_csv('Shops.csv')

Answer 1

使用 isin 进行过滤，然后对“D”执行 groupby 并迭代保存到 CSV。

incl = ['Home', 'Away', 'Shops']    
for k, g in df[df['D'].isin(incl)].groupby('D'):
    g.to_csv(f'{k}.csv')  # '{}.csv'.format(k)

当且仅当您的类别多于要保存的类别时，isin 过滤步骤才重要。如果情况并非如此，并且您想保存所有内容，您的解决方案将会简化:

for k, g in df.groupby('D'):
    ...