Использование groupby (). Sum () на фрейме данных, а затем построение круговой диаграммы с метками?

Question

Это мой первый вопрос, я довольно новичок в Python / pandas / matplolib

У меня есть эта строка кода, которая создает DataFrame:

repartition = sorted2017.groupby(by=sorted2017["Traitement"]).sum()

Он работает какЯ ожидал, за исключением того, что заголовок столбца «Traitement», кажется, появляется в отдельной строке:

                  Prix  Coût net  Manuvie     CCQ  SSQ
Traitement                                            
masso (Véro)    213.86       0.0    144.0   69.86  0.0
ostéo (Véro)     80.00       0.0     64.0   16.00  0.0
physio (Danny)  415.00       0.0    265.0  150.00  0.0
physio (Véro)   269.00       0.0    204.8   64.20  0.0
psy (Simone)    500.00       0.0    150.0  350.00  0.0
psy (Véro)      300.00       0.0    240.0   60.00  0.0

Я хотел использовать столбец «Traitement» в качестве меток для моей круговой диаграммы matplotlib, поэтому я попытался:

plt.pie(repartition["Prix"], labels=repartition["Traitement"])

plt.show ()

Но я получаю KeyError.Я также пытался использовать iloc для меток, но затем я получаю

ValueError: "'label' должен иметь длину 'x'"

Как я могуисправить это?

Answer 1

После groupby, столбец "Traitement" находится в столбце индекса.

plt.pie(x=repartition["Prix"], labels=repartition.index)
plt.show()