Как сделать свод в пандах с дублированными записями - PullRequest
Новая
       9

Как сделать свод в пандах с дублированными записями

0 голосов
/ 04 октября 2019

Как можно сделать поворот в пандах? Я не могу обойти ошибку «повторяющиеся записи». Вход и выход должны выглядеть так, как показано ниже. 

import pandas as pd

input = pd.DataFrame({'measure': ['length','length','length','weight','weight','weight','sex','sex','sex'],
                      'species': [10, 10, 10, 10, 10, 10, 10, 10, 10],
                      'value': [1, 2, 3, 13, 45, 123, 0, 1, 1],
                      'set': [3, 3, 3, 3, 3, 3, 3, 3, 3]})

output = pd.DataFrame({'set': [3,3,3],
                       'species': [10, 10, 10],
                       'length': [1, 2, 3],
                       'weight': [13, 45, 123],
                       'sex': [0, 1, 1]})

test = input.pivot(index='set',columns='measure', values='value')

print(test)

1 Ответ

5 голосов
/ 04 октября 2019

В этой ситуации мы обычно решаем groupby().cumcount(), чтобы получить новый индекс: 

indf['idx'] = indf.groupby('measure').cumcount()
(indf.pivot_table(index=['idx','species','set'], 
                 columns='measure', 
                 values='value')
     .reset_index(('species','set'))
)

Вывод: 

measure  species  set  length  sex  weight
idx                                       
0             10    3       1    0      13
1             10    3       2    1      45
2             10    3       3    1     123
...