Pandas: como agrupar por índice e realizar um cálculo

Você pode usar os seguintes métodos para agrupar uma ou mais colunas de índice em pandas e realizar cálculos:

Método 1: agrupar por uma coluna de índice

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

Método 2: agrupar por múltiplas colunas de índice

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

Método 3: agrupar por coluna de índice e coluna regular

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

Os exemplos a seguir mostram como usar cada método com o seguinte DataFrame do pandas que possui um 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

Método 1: agrupar por uma coluna de índice

O código a seguir mostra como encontrar o valor máximo da coluna “pontos”, agrupados pela coluna do índice “posição”:

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

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

Método 2: agrupar por múltiplas colunas de índice

O código a seguir mostra como encontrar a soma da coluna “pontos”, agrupada pelas colunas de índice “equipe” e “posição”:

 #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

Método 3: agrupar por coluna de índice e coluna regular

O código a seguir mostra como encontrar o número de valores exclusivos na coluna “rebotes”, agrupados pela coluna de índice “equipe” e pela coluna regular de “pontos”:

 #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

Recursos adicionais

Os tutoriais a seguir explicam como realizar outras operações comuns em pandas:

Como contar valores únicos em pandas
Como nivelar MultiIndex em Pandas
Como modificar um ou mais valores de índice no Pandas
Como redefinir um índice no Pandas