Pandas pd.pivot_table выдает KeyError

Question

У меня есть df фрейм данных с 4 столбцами 'year', 'cath1', 'cath2' и 'cath3' и 2000 записей, соответствующих продуктам, с соответствующим годом производства и значением в каждой 3 категории.

Я хотел бы создать еще один фрейм данных с теми же столбцами 3 категорий и вычислить среднее значение всех продуктов для каждого конкретного года в каждой из этих категорий.

Я пытался использовать следующий код, но онне работает.

df1=pd.pivot_table(df,index=['year'],values=['0','1','2'],aggfunc=np.mean)

Exception has occurred: KeyError '0'

Answer 1

Ваш код правильный, но вам нужно использовать имя столбца в значениях.

df_pivot = pd.pivot_table(df, index=['year'], values=['cath1', 'cath2','cath3'],
                          aggfunc=np.mean)