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'))
即可得到各自输出文件。