El marco de datos de Pandas agrega un campo basado en múltiples declaraciones if
Soy bastante nuevo en Python y Pandas, por lo que esta podría ser una pregunta obvia.
Tengo un marco de datos con edades enumeradas en él. Quiero crear un nuevo campo con una banda de edad. Puedo usar la declaración lambda para capturar una sola declaración if / else, pero quiero usar varios if, p.if age < 18 then 'under 18' elif age < 40 then 'under 40' else '>40'.
No creo que pueda hacer esto usando lambda, pero no estoy seguro de cómo hacerlo de una manera diferente. Tengo este código hasta ahora:
import pandas as pd
import numpy as n
d = {'Age' : pd.Series([36., 42., 6., 66., 38.]) }
df = pd.DataFrame(d)
df['Age_Group'] = df['Age'].map(lambda x: '<18' if x < 19 else '>18')
print(df)