Pandas 按照一列中所有值进行筛选并导出至单独文件

Pandas 按照一列中所有值进行筛选并导出至单独文件

@author: SUFEHeisenberg
@date: 2022/10/03

数据示例

import pandas as pd
df = pd.DataFrame({"Key":['C','B','C','A','B','B','A','C','A'],
                           "Data":[2,4,6,8,10,1,14,16,18]})
# In [2]: df
#Out[2]:
#  Key  Data
#0   C     2
#1   B     4
#2   C     6
#3   A     8
#4   B    10
#5   B     1
#6   A    14
#7   C    16
#8   A    18

处理代码

data.groupby('key',axis=0).apply(lambda d: d.to_csv(f'{d.name}.csv'))

即可得到各自输出文件。