Hoe tel je het aantal werknemers per salariscategorieën met behulp van Pandas DataFrame?

Om het aantal werknemers per salariscategorie te tellen in een Pandas DataFrame, kun je de methode groupby gebruiken samen met size of count. Stel dat je een DataFrame hebt met de kolommen employee_id en salary_category.

Voorbeeld:

import pandas as pd

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

# Aantal werknemers per salariscategorie
counts = df.groupby('salary_category').size()
print(counts)

De output zal zijn:

salary_category
hoog      2
laag       2
medium    2
dtype: int64

Wil je het resultaat als een DataFrame met een benoemde kolom:

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

Dit is handig voor verdere analyse of visualisatie.