Hogyan számolhatjuk meg a munkavállalók számát a fizetési kategóriák szerint Pandas DataFrame segítségével?

Az alkalmazottak számának számlálásához fizetési kategóriák szerint egy Pandas DataFrame-ben a groupby módszert használhatod a size vagy count mellett. Tegyük fel, hogy van egy DataFrame-ed employee_id és salary_category oszlopokkal.

Példa:

import pandas as pd

data = {
    'employee_id': [1, 2, 3, 4, 5, 6],
    'salary_category': ['alacsony', 'közepes', 'közepes', 'magas', 'alacsony', 'magas']
}
df = pd.DataFrame(data)

# Az alkalmazottak száma fizetési kategóriánként
számok = df.groupby('salary_category').size()
print(számok)

A kimenet:

salary_category
magas      2
alacsony   2
közepes    2
dtype: int64

Ha eredményt szeretnél egy DataFrame-ben, ahol a oszlop neve például employee_count, akkor:

számok = df.groupby('salary_category').size().reset_index(name='employee_count')
print(számok)

Ez hasznos a további elemzésekhez vagy vizualizációkhoz.