Для цикла после панды группового - PullRequest
Новая
       25

Для цикла после панды группового

0 голосов
/ 19 декабря 2018

Мой входной фрейм данных выглядит следующим образом: 

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

1 Ответ

0 голосов
/ 19 декабря 2018

Это сработало: 

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

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