Результаты отображения pandas, сгруппированные по столбцу

Question

Я пытаюсь по существу сделать то же самое, что и сводная таблица в Excel, но с использованием pandas. Вот некоторые из моих данных:

First_Name   Last_Name Country Prize_Money 
     Roger    Federer  SUI 88691538 
     Novak    Djokovic  SRB 72444493 
     Rafael    Nadal  ESP 71421511 
     Andy    Murray  GBR 34190085 
     David    Ferrer  ESP 24928109 

Выход будет страна и сумма колонки Prize_Money:

Country Sum_Prize_Money 
    ESP  96349620 
    GBR  34190085 

и так далее.

Я создал объект DataFrame с данными:

df = pd.DataFrame(Analysis_ATP_data_list, columns=['Ranking', 
                'First_Name', 
                'Last_Name', 
                'Country', 
                'Prize_Money']) 

и хотел бы напечатать резюме grouped_by на экране.

Спасибо заранее,

Answer 1

Вы можете позвонить groupby:

>>> df.groupby("Country")["Prize_Money"].sum() 
Country 
ESP  96349620 
GBR  34190085 
SRB  72444493 
SUI  88691538 
Name: Prize_Money, dtype: int64 

если вы хотите серии, как выход, или, может быть,

>>> df.groupby("Country", as_index=False)["Prize_Money"].sum() 
    Country Prize_Money 
0  ESP  96349620 
1  GBR  34190085 
2  SRB  72444493 
3  SUI  88691538 

если вы хотите кадр. Чтение документов на grouping, вероятно, хорошая идея.

---

Более сложные агрегаты возможны также:

>>> df.groupby("Country", as_index=False).agg({"Prize_Money": ["sum", "count"]}) 
    Country Prize_Money  
        sum count 
0  ESP 96349620  2 
1  GBR 34190085  1 
2  SRB 72444493  1 
3  SUI 88691538  1