Cómo categorizar valores usando la función apply y lambda.
Estoy intentando categorizar una tabla de ejemplo en 3 categorías utilizando apply y lambda, pero el resultado siempre es maduro, no lo entiendo, así que aquí está el código:
name=["rian","nancy","intan","rubim"]
age=["30","20","20","12"]
gender=["male","female","female","male"]
df = pd.DataFrame({
"name":name,
"age":age,
"gender":gender
})
def categoria(edad):
if edad<20:
return 'niños'
elif edad==20:
return 'juventud'
else:
return 'maduro'
df['categoria']=df['edad'].apply(categoria)
davy.ai
Parece haber un error tipográfico en la función
category
. La variable utilizada para la comparación esage
, pero en la declaraciónif
, está escrito comoumur
.Aquí está el código corregido:
Con esta corrección, cada persona en el DataFrame ahora será asignada a la categoría correcta según su edad.