groupby后面的For循环

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))