Как найти медианное значение на группу в пандах
Вы можете использовать следующий базовый синтаксис для расчета медианного значения для каждой группы в пандах:
df. groupby ([' group_variable '])[' value_variable ']. median (). reset_index ()
Вы также можете использовать следующий синтаксис для расчета медианного значения, сгруппированного по нескольким столбцам:
df. groupby ([' group1 ', ' group2 '])[' value_variable ']. median (). reset_index ()
Следующие примеры показывают, как использовать этот синтаксис на практике.
Пример 1. Поиск медианного значения по группе
Предположим, у нас есть следующие DataFrames 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
Мы можем использовать следующий код, чтобы найти медианное значение столбца «очки», сгруппированное по командам:
#calculate median points by team
df. groupby ([' team '])[' points ']. median (). reset_index ()
team points
0 to 7.0
1 B 9.0
По результату мы видим:
- Среднее количество очков, набранных игроками команды А, равно 7 .
- Среднее количество очков, набранных игроками команды Б, составляет 9 .
Обратите внимание, что мы также можем найти медианное значение сразу двух переменных:
#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
Пример 2. Поиск медианного значения по нескольким группам
Следующий код показывает, как найти медианное значение столбца «очки», сгруппированное по команде и позиции:
#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
По результату мы видим:
- Среднее количество очков, набранных игроками на позиции «F» в команде А, составляет 8 .
- Медиана очков, набранных игроками позиции «G» команды А, равна 6 .
- Среднее количество очков, набранных игроками на позиции «F» в команде B, составляет 6,5 .
- Медиана очков, набранных игроками позиции «G» команды Б, равна 10,5 .
Дополнительные ресурсы
В следующих руководствах объясняется, как выполнять другие распространенные функции в pandas:
Как найти максимальное значение для группы в Pandas
Как найти сумму на группу в пандах
Как рассчитать квантили по группе в Pandas