Pandas: як групувати за індексом і виконувати обчислення

Ви можете використовувати наступні методи для групування за одним або декількома стовпцями індексу в pandas і виконання обчислень:

Спосіб 1: групування за стовпцем індексу

 df. groupby (' index1 ')[' numeric_column ']. max ()

Спосіб 2: групування за кількома стовпцями індексу

 df. groupby ([' index1 ',' index2 '])[' numeric_column ']. sum ()

Спосіб 3: групування за стовпцем індексу та звичайним стовпцем

 df. groupby ([' index1 ',' numeric_column1 '])[' numeric_column2 ']. nunique ()

У наведених нижче прикладах показано, як використовувати кожен метод із наступним pandas DataFrame, який має MultiIndex:

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' team ': ['A', 'A', 'A', 'A', 'A', 'B', 'B', 'B', 'B', 'B'],
                   ' position ': ['G', 'G', 'G', 'F', 'F', 'G', 'G', 'F', 'F', 'F'],
                   ' points ': [7, 7, 7, 19, 16, 9, 10, 10, 8, 8],
                   ' rebounds ': [8, 8, 8, 10, 11, 12, 13, 13, 15, 11]})

#set 'team' column to be index column
df. set_index ([' team ', ' position '], inplace= True )

#view DataFrame
df

		 rebound points
team position		
A G 7 8
        G 7 8
        G 7 8
        F 19 10
        F 16 11
B G 9 12
        G 10 13
        F 10 13
        F 8 15
        F 8 11

Спосіб 1: групування за стовпцем індексу

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

 #find max value of 'points' grouped by 'position index column
df. groupby (' position ')[' points ']. max ()

position
F 19
G 10
Name: points, dtype: int64

Спосіб 2: групування за кількома стовпцями індексу

У наведеному нижче коді показано, як знайти суму стовпця «очки», згруповану за стовпцями індексу «команда» та «позиція».

 #find max value of 'points' grouped by 'position index column
df. groupby ([' team ', ' position '])[' points ']. sum ()

team position
AF35
      G21
BF 26
      G 19
Name: points, dtype: int64

Спосіб 3: групування за стовпцем індексу та звичайним стовпцем

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

 #find max value of 'points' grouped by 'position index column
df. groupby ([' team ', ' points '])[' rebounds ']. nunique ()

team points
At 7 1
      16 1
      19 1
B 8 2
      9 1
      10 1
Name: rebounds, dtype: int64

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

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

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