python pandas或其它,将一个csv按一列的id值切成多个csv,并命名成id.csv

Question

如下图,一个订单里面有多个order id,需要按order id分成多个,并命名成order id值,每个切割后的表都要有表头那一行.
原表如图

切割后如图,

切割后变成三个, 033505935, 033505936, 033505937,并且原表需要保留
麻烦指教一下,手动粘贴太麻烦了

Answer 1

import pandas as pd

content = pd.read_csv('test.csv')
filename = list(set(content.iloc[:,0]))   #测试第一列,索引号0

for f in filename:
    txt = content.loc[content['a']==f]  #第一列的名字为'a'
    txt.to_csv( f + '.csv',index=False)

测试文件test.csv

a,b,c,d
id1,1,2,3
id1,2,2,2
id2,1,1,1
id1,3,3,3
id2,4,4,4
id3,1,1,1
id3,2,2,2
id1,5,5,5