Cum se calculează numărul de angajați pe categorii de salarii folosind Pandas DataFrame?

Pentru a număra numărul de angajați pe categorii de salariu într-un DataFrame Pandas, poți folosi metoda groupby împreună cu size sau count. Presupunem că ai un DataFrame cu coloanele employee_id și salary_category.

Exemplu:

import pandas as pd

date = {
    'employee_id': [1, 2, 3, 4, 5, 6],
    'salary_category': ['scăzut', 'mediu', 'mediu', 'ridicat', 'scăzut', 'ridicat']
}
df = pd.DataFrame(date)

# Numărul de angajați în fiecare categorie de salariu
numere = df.groupby('salary_category').size()
print(numere)

Ieșirea va fi:

salary_category
ridicat     2
scăzut      2
mediu       2
dtype: int64

Dacă vrei să obții rezultatul sub formă de DataFrame cu o coloană denumită:

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

Acest lucru este util pentru analize ulterioare sau vizualizări.