Comment compter le nombre d'employés par catégories de salaire avec Pandas DataFrame?

Pour compter le nombre d'employés par catégories de salaire dans un DataFrame Pandas, vous pouvez utiliser la méthode groupby avec size ou count. Supposons que vous avez un DataFrame avec les colonnes employee_id et salary_category.

Exemple:

import pandas as pd

donnees = {
    'employee_id': [1, 2, 3, 4, 5, 6],
    'salary_category': ['faible', 'moyen', 'moyen', 'élevé', 'faible', 'élevé']
}
df = pd.DataFrame(donnees)

# Compter le nombre d'employés dans chaque catégorie de salaire
comptes = df.groupby('salary_category').size()
print(comptes)

La sortie sera:

salary_category
élevé      2
faible     2
moyen      2
dtype: int64

Si vous souhaitez obtenir le résultat sous forme de DataFrame avec une colonne nommée:

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

Ceci est pratique pour une analyse ou une visualisation ultérieure.