DataFrame.groupby()
Группирует строки для агрегации по категориям.
Сигнатура
DataFrame.groupby(by, as_index=True)Разбивает таблицу на группы по значениям столбца, после чего к каждой группе применяют агрегирующую функцию.
import pandas as pd
df = pd.DataFrame({'отдел': ['IT', 'IT', 'HR'], 'зп': [100, 120, 80]})
print(df.groupby('отдел')['зп'].mean())
# отдел
# HR 80.0
# IT 110.0