Перейти к содержимому
Задача · NumPy и pandas · урок 12 из 12

Задача: медиана по группам

Напишите функцию group_medians(df, by, column): она группирует строки таблицы df по столбцу by и возвращает словарь. Ключ словаря — значение из by, значение — медиана столбца column в этой группе.

  • df = pd.DataFrame({"team": ["a", "a", "a", "b"], "salary": [100, 400, 200, 50]}), group_medians(df, "team", "salary") → {"a": 200.0, "b": 50.0}
  • те же команды, "salary": [100, np.nan, 120, 80] → {"a": 110.0, "b": 80.0}
  • "team": ["a", "a", "b", "b"], "salary": [100, 120, np.nan, np.nan] → {"a": 110.0}

Ограничения: в столбце by пропусков нет, а в column — числа и пропуски; при чётном числе значений медиана — среднее двух центральных; пропуски не учитываются, а группа без единого значения в словарь не попадает.

Не добавляйте лишних импортов, если в задаче это прямо не сказано.

group_medians.py — python
python · строк: 6
Ctrl+Enter запуск · Esc выйти из редактора

Подпишись, чтобы пройти курс

Курс бесплатный. Войди через Telegram и подпишись на канал — откроются все уроки.

Войти другим способом