Задача · 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 — числа и пропуски; при чётном числе значений медиана — среднее двух центральных; пропуски не учитываются, а группа без единого значения в словарь не попадает.
Не добавляйте лишних импортов, если в задаче это прямо не сказано.
Подпишись, чтобы пройти курс
Курс бесплатный. Войди через Telegram и подпишись на канал — откроются все уроки.