Я новичок в Python и Pandas, так что это может быть очевидный вопрос.
У меня есть датафрейм с возрастом, указанным в нем. Я хочу создать новое поле с возрастной полосой. Я могу использовать оператор lambda для захвата одного оператора if/else, но я хочу использовать несколько, если, например, if age < 18 then 'under 18' elif age < 40 then 'under 40' else '>40'
.
Я не думаю, что могу это сделать, используя лямбда, но не уверен, как это сделать по-другому. У меня есть этот код:
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)