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)]