Как подсчитать количество сотрудников по категориям зарплаты с помощью Pandas DataFrame?

Для подсчёта количества сотрудников по категориям зарплаты в Pandas DataFrame можно использовать метод groupby вместе с size или count. Предположим, у вас есть DataFrame с колонками employee_id и salary_category.

Пример:

import pandas as pd

data = {
    'employee_id': [1, 2, 3, 4, 5, 6],
    'salary_category': ['low', 'medium', 'medium', 'high', 'low', 'high']
}
df = pd.DataFrame(data)

# Подсчёт количества сотрудников в каждой категории зарплаты
counts = df.groupby('salary_category').size()
print(counts)

Вывод будет:

salary_category
high      2
low       2
medium    2
dtype: int64

Если нужно получить результат в виде DataFrame с именованным столбцом:

counts = df.groupby('salary_category').size().reset_index(name='employee_count')
print(counts)

Это удобно для дальнейшего анализа или визуализации.