Kā saskaitīt darbinieku skaitu pēc algu kategorijām, izmantojot Pandas DataFrame?

Lai skaitītu darbinieku skaitu pēc algas kategorijām Pandas DataFrame, jūs varat izmantot metodi groupby kopā ar size vai count. Pieņemsim, ka jums ir DataFrame ar kolonnām employee_id un salary_category.

Piemērs:

import pandas as pd

dati = {
    'employee_id': [1, 2, 3, 4, 5, 6],
    'salary_category': ['zems', 'vidējs', 'vidējs', 'augsts', 'zems', 'augsts']
}
df = pd.DataFrame(dati)

# Darbinieku skaits pēc algas kategorijām
skaitļi = df.groupby('salary_category').size()
print(skaitļi)

Rezultāts būs:

salary_category
augsts     2
zems       2
vidējs     2
dtype: int64

Ja vēlaties iegūt rezultātu kā DataFrame ar nosaukto kolonnu:

skaitļi = df.groupby('salary_category').size().reset_index(name='employee_count')
print(skaitļi)

Tas ir noderīgi turpmākai analīzei vai vizualizācijai.