Як виконати суму groupby у pandas (з прикладами)

Ви можете використовувати такий базовий синтаксис, щоб знайти суму значень за групою в pandas:

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

Наступні приклади показують, як використовувати цей синтаксис на практиці з наступними pandas DataFrame:

 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

Приклад 1: згрупувати за стовпцем, додати стовпець

Наступний код показує, як групувати за стовпцями та підсумовувати значення в стовпці:

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

	team points
0 to 65
1 B 31

З результату ми бачимо, що:

• У сумі гравці команди А набрали 65 очок.
• Гравці команди Б набрали 31 очко.

Приклад 2: групування за кількома стовпцями, додавання кількох стовпців

Наступний код показує, як згрупувати кілька стовпців і підсумувати кілька стовпців:

 #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

З результату ми бачимо, що:

• Гравці команди А на позиції «С» набрали 9 очок і 6 підбирань.
• Гравці команди А на позиції «F» набрали 14 очок і 10 підбирань.
• Гравці команди А на позиції «G» загалом набрали 42 очки та 19 підбирань.

І так далі.

Зауважте, що функція reset_index() запобігає групуванню стовпців у складі індексу.

Наприклад, ось як виглядає результат, якщо ми його не використовуємо:

 #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

Залежно від того, як ви хочете, щоб результати виглядали, ви можете або не можете використовувати функцію reset_index() .

Додаткові ресурси

У наступних посібниках пояснюється, як виконувати інші типові операції групування в pandas:

Як підрахувати групові спостереження панд
Як знайти максимальне значення для групи в Pandas
Як обчислити квантилі за групами в Pandas