Como contar o número de funcionários por categorias salariais usando Pandas DataFrame?

Para contar o número de funcionários por categorias de salário num DataFrame do Pandas, pode usar o método groupby juntamente com size ou count. Suponha que tem um DataFrame com as colunas employee_id e salary_category.

Exemplo:

import pandas as pd

dados = {
    'employee_id': [1, 2, 3, 4, 5, 6],
    'salary_category': ['baixo', 'médio', 'médio', 'alto', 'baixo', 'alto']
}
df = pd.DataFrame(dados)

# Contar o número de funcionários em cada categoria salarial
contagens = df.groupby('salary_category').size()
print(contagens)

A saída será:

salary_category
alto      2
baixo     2
médio     2
dtype: int64

Se desejar obter o resultado como um DataFrame com uma coluna nomeada:

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

Isto é útil para análises ou visualizações futuras.