Задача курса

groupby и transform: групповые признаки на уровне строк и фильтрация групп — задача по NumPy и pandas

Курс «NumPy и pandas: практический тренажёр» · урок «groupby и transform: групповые признаки на уровне строк и фильтрация групп»

Условие

DataFrame df содержит столбцы 'dept' и 'salary'. Нужно оставить только строки сотрудников из отделов, где суммарная зарплата превышает 200000. Какой подход использует transform('sum'), возвращает Series с тем же индексом и длиной, что у df, и позволяет сразу передать булеву маску в df.loc?

Попробуйте решить

Выберите один вариантmask = df.groupby('dept')['salary'].transform('sum') > 200000; df.loc[mask]agg_df = df.groupby('dept')['salary'].agg('sum') > 200000; df.loc[agg_df]mask = df.groupby('dept')['salary'].transform('mean') > 200000; df.loc[mask]df[df.groupby('dept')['salary'].agg('sum').reindex(df['dept']).values > 200000]totals = df.groupby('dept')['salary'].agg('sum'); df[df['dept'].isin(totals[totals > 200000].index)]

Без регистрации · результат не сохраняется