Ako spočítať počet zamestnancov podľa kategórií platov pomocou Pandas DataFrame?

Pre spočítanie počtu zamestnancov podľa kategórií platu v DataFrame Pandas môžete použiť metódu groupby spolu s size alebo count. Predpokladáme, že máte DataFrame s stĺpcami employee_id a salary_category.

Príklad:

import pandas as pd

dáta = {
    'employee_id': [1, 2, 3, 4, 5, 6],
    'salary_category': ['nízky', 'stredný', 'stredný', 'vysoký', 'nízky', 'vysoký']
}
df = pd.DataFrame(dáta)

# Počet zamestnancov v každej kategórii platu
počty = df.groupby('salary_category').size()
print(počty)

Výstup bude:

salary_category
vysoký     2
nízky      2
stredný    2
dtype: int64

Ak chcete získať výsledok vo forme DataFrame s pomenovaným stĺpcom:

počty = df.groupby('salary_category').size().reset_index(name='employee_count')
print(počty)

Toto je užitočné pre ďalšiu analýzu alebo vizualizáciu.