Jak spočítat počet zaměstnanců podle kategorií platů pomocí Pandas DataFrame?

Pro spočítání počtu zaměstnanců podle kategorií platů v DataFrame Pandas můžete použít metodu groupby spolu s size nebo count. Předpokládejme, že máte DataFrame s sloupci employee_id a salary_category.

Příklad:

import pandas as pd

data = {
    'employee_id': [1, 2, 3, 4, 5, 6],
    'salary_category': ['nízký', 'střední', 'střední', 'vysoký', 'nízký', 'vysoký']
}
df = pd.DataFrame(data)

# Počet zaměstnanců v každé kategorii platu
pocty = df.groupby('salary_category').size()
print(pocty)

Výstup bude:

salary_category
vysoký     2
nízký      2
střední    2
dtype: int64

Pokud chcete výsledek ve formě DataFrame s pojmenovaným sloupcem:

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

Toto je užitečné pro další analýzy nebo vizualizace.