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)

Ответы на вопрос(2)

Ваш ответ на вопрос