Kuidas arvutada töötajate arv palgakategooriate kaupa, kasutades Pandas DataFrame?

Selleks, et Pandas DataFrame'is töötajate arvu arvutada vastavalt palgakategooriatele, saate kasutada meetodit groupby koos size või count. Oletame, et teil on DataFrame, millel on veerud employee_id ja salary_category.

Näide:

import pandas as pd

data = {
    'employee_id': [1, 2, 3, 4, 5, 6],
    'salary_category': ['madal', 'keskmine', 'keskmine', 'kõrge', 'madal', 'kõrge']
}
df = pd.DataFrame(data)

# Töötajate arv iga palgakategooria kohta
arvud = df.groupby('salary_category').size()
print(arvud)

Tulemus on:

salary_category
kõrge      2
madal       2
keskmine    2
dtype: int64

Kui soovite saada tulemuse DataFrame'ina koos nimetatud veeruga:

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

See on kasulik edasiseks analüüsiks või visualiseerimiseks.