熊猫 groupby 之后的 for 循环答案

【问题标题】：For loop after pandas groupby熊猫 groupby 之后的 for 循环
【发布时间】：2018-12-19 20:23:47
【问题描述】：

我的输入数据框如下所示：

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

【问题讨论】：

groupby 和 csv
做：group.to_csv('/test/output_{}.csv'.format(group['uniqueID'][0]))
addData['uniqueID'][0] 将永远是一个单一的值，0001。因此output_csv 将永远是那个值

标签： python python-3.x pandas for-loop group-by

【解决方案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))

【讨论】：