Question

我的输入数据框如下：

name   uniqueID
kate   0001
sam    0001
lucy   0002
wes    0001
kip    0002

我有以下内容：

addData =pd.read_csv('/input.csv')
grouped = addData.groupby(['uniqueID'])
filename = addData['uniqueID'][0]
output_csv = '/test/output_{}.csv'.format(filename)

for name, group in grouped:
    group.to_csv(output_csv)

我的输出是半正确的。我有一个文件，其中包含'uniqueID'，EX）output001.csv的所有相关记录：

name   uniqueID
kate   0001
sam    0001
wes    0001

问题是我只得到一个文件 -我的循环无法正常产生output0001.csv和output0002.csv

Answer 1

这有效：

grouped = addData.groupby(['uniqueID'])
filename = addData['uniqueID'][0]
output_csv = 'output_{}.csv'

for name, group in grouped:
    group.to_csv(output_csv.format(name))

熊猫分组后循环

1 个答案: