Como encontrar o valor mediano por grupo em pandas
Você pode usar a seguinte sintaxe básica para calcular o valor mediano por grupo em pandas:
df. groupby ([' group_variable '])[' value_variable ']. median (). reset_index ()
Você também pode usar a seguinte sintaxe para calcular o valor mediano, agrupado por várias colunas:
df. groupby ([' group1 ', ' group2 '])[' value_variable ']. median (). reset_index ()
Os exemplos a seguir mostram como usar essa sintaxe na prática.
Exemplo 1: Encontrando o valor mediano por um grupo
Vamos supor que temos os seguintes DataFrames do pandas:
import pandas as pd #createDataFrame df = pd. DataFrame ({' team ': ['A', 'A', 'A', 'A', 'B', 'B', 'B', 'B'], ' position ': ['G', 'G', 'F', 'F', 'G', 'G', 'F', 'F'], ' points ': [5, 7, 7, 9, 12, 9, 9, 4], ' rebounds ': [11, 8, 10, 6, 6, 5, 9, 12]}) #view DataFrame df team position points rebounds 0 A G 5 11 1 A G 7 8 2 A F 7 10 3 A F 9 6 4 B G 12 6 5 B G 9 5 6 B F 9 9 7 B F 4 12
Podemos usar o seguinte código para encontrar o valor mediano da coluna “pontos”, agrupados por equipe:
#calculate median points by team
df. groupby ([' team '])[' points ']. median (). reset_index ()
team points
0 to 7.0
1 B 9.0
Pelo resultado podemos ver:
- A mediana de pontos marcados pelos jogadores da equipe A é 7 .
- A mediana de pontos marcados pelos jogadores da equipe B é 9 .
Observe que também podemos encontrar o valor mediano de duas variáveis ao mesmo tempo:
#calculate median points and median rebounds by team
df. groupby ([' team '])[[' points ', ' rebounds ']]. median ()
team points rebounds
0 to 7.0 9.0
1B 9.0 7.5
Exemplo 2: Encontrando o valor mediano por vários grupos
O código a seguir mostra como encontrar o valor mediano da coluna “pontos”, agrupados por equipe e posição:
#calculate median points by team
df. groupby ([' team ', ' position '])[' points ']. median (). reset_index ()
team position points
0 A F 8.0
1 A G 6.0
2 B F 6.5
3 B G 10.5
Pelo resultado podemos ver:
- A mediana de pontos marcados pelos jogadores na posição “F” da equipe A é 8 .
- A mediana de pontos marcados pelos jogadores da posição “G” da equipe A é 6 .
- A mediana de pontos marcados pelos jogadores da posição “F” da equipe B é de 6,5 .
- A mediana de pontos marcados pelos jogadores da posição “G” da equipe B é 10,5 .
Recursos adicionais
Os tutoriais a seguir explicam como executar outras funções comuns em pandas:
Como encontrar o valor máximo por grupo no Pandas
Como encontrar a soma por grupo em pandas
Como calcular quantis por grupo no Pandas