Sobes.tech
Middle

¿Cómo contar el número de empleados por categorías de salario usando Pandas DataFrame?

sobes.tech AI

Respuesta de la IA

Para contar la cantidad de empleados por categorías de salario en un DataFrame de Pandas, puedes usar el método groupby junto con size o count. Supón que tienes un DataFrame con las columnas employee_id y salary_category.

Ejemplo:

import pandas as pd

datos = {
    'employee_id': [1, 2, 3, 4, 5, 6],
    'salary_category': ['bajo', 'medio', 'medio', 'alto', 'bajo', 'alto']
}
df = pd.DataFrame(datos)

# Contar la cantidad de empleados en cada categoría salarial
cuentas = df.groupby('salary_category').size()
print(cuentas)

La salida será:

salary_category
alto      2
bajo      2
medio     2
dtype: int64

Si deseas obtener el resultado en forma de DataFrame con una columna nombrada:

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

Esto es útil para análisis o visualización posterior.