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.