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