Come contare il numero di dipendenti per categorie di stipendio usando Pandas DataFrame?

Per contare il numero di dipendenti per categorie di salario in un DataFrame di Pandas, puoi usare il metodo groupby insieme a size o count. Supponi di avere un DataFrame con le colonne employee_id e salary_category.

Esempio:

import pandas as pd

dati = {
    'employee_id': [1, 2, 3, 4, 5, 6],
    'salary_category': ['basso', 'medio', 'medio', 'alto', 'basso', 'alto']
}
df = pd.DataFrame(dati)

# Contare il numero di dipendenti in ogni categoria di salario
conta = df.groupby('salary_category').size()
print(conta)

L'output sarà:

salary_category
alto      2
basso     2
medio     2
dtype: int64

Se vuoi ottenere il risultato come DataFrame con una colonna nominata:

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

Questo è utile per analisi o visualizzazioni successive.