Jak wykonać sumę groupby w pandach (z przykładami)

Aby znaleźć sumę wartości według grup w pandach, możesz użyć następującej podstawowej składni:

 df. groupby ([' group1 ',' group2 '])[' sum_col ']. sum (). reset_index ()

Poniższe przykłady pokazują, jak w praktyce używać tej składni z następującą ramką DataFrame pand:

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' team ': ['A', 'A', 'A', 'A', 'B', 'B', 'B', 'B'],
                   ' position ': ['G', 'G', 'F', 'C', 'G', 'F', 'F', 'C'],
                   ' points ': [25, 17, 14, 9, 12, 9, 6, 4],
                   ' rebounds ': [11, 8, 10, 6, 6, 5, 9, 12]})

#view DataFrame
df

	team position points rebounds
0 A G 25 11
1 A G 17 8
2 A F 14 10
3 A C 9 6
4 B G 12 6
5 B F 9 5
6 B F 6 9
7 B C 4 12

Przykład 1: grupuj według kolumny, dodaj kolumnę

Poniższy kod pokazuje, jak grupować według kolumn i sumować wartości w kolumnie:

 #group by team and sum the points
df. groupby ([' team '])[' points ']. sum (). reset_index ()

	team points
0 to 65
1 B 31

Z wyniku możemy zobaczyć, że:

• Zawodnicy drużyny A zdobyli łącznie 65 punktów.
• Zawodnicy drużyny B zdobyli łącznie 31 punktów.

Przykład 2: grupuj według wielu kolumn, dodaj wiele kolumn

Poniższy kod pokazuje, jak grupować wiele kolumn i sumować wiele kolumn:

 #group by team and position, sum points and rebounds
df. groupby ([' team ', ' position '])[' points ', ' rebounds ']. sum (). reset_index ()

        team position points rebounds
0 A C 9 6
1 A F 14 10
2 A G 42 19
3 B C 4 12
4 B F 15 14
5 B G 12 6

Z wyniku możemy zobaczyć, że:

• Zawodnicy Drużyny A na pozycji „C” zdobyli łącznie 9 punktów i 6 zbiórek.
• Zawodnicy Drużyny A na pozycji „F” zdobyli łącznie 14 punktów i 10 zbiórek.
• Zawodnicy Drużyny A na pozycji „G” zdobyli łącznie 42 punkty i 19 zbiórek.

I tak dalej.

Należy pamiętać, że funkcja reset_index() zapobiega grupowaniu kolumn jako części indeksu.

Na przykład tak wygląda wynik, jeśli go nie użyjemy:

 #group by team and position, sum points and rebounds
df. groupby ([' team ', ' position '])[' points ', ' rebounds ']. sum ()

                 rebound points
team position		
A C 9 6
F 14 10
G 42 19
B C 4 12
F 15 14
G 12 6

W zależności od tego, jak mają wyglądać wyniki, możesz użyć funkcji reset_index() lub nie.

Dodatkowe zasoby

Poniższe samouczki wyjaśniają, jak wykonywać inne typowe operacje grupowania w pandach:

Jak liczyć obserwacje grupowe u pand
Jak znaleźć maksymalną wartość na grupę w Pandas
Jak obliczyć kwantyle według grupy w Pandach