Панды: как группировать по индексу и выполнять расчет

Вы можете использовать следующие методы для группировки по одному или нескольким столбцам индекса в 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 ()

В следующих примерах показано, как использовать каждый метод со следующим DataFrame pandas, имеющим 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. Группировка по индексному столбцу.

Следующий код показывает, как найти максимальное значение столбца «пункты», сгруппированное по столбцу индекса «позиция»:

 #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