Create a new column “age_group”. This column should contain the age group the respondent belongs to. The age groups are 18-24, 25-40, 41-55 and 55+

Question

I know it can be done using map() function just don't know how. I have tried the below code but it's not working and showing NaN as output.

df['age_group'] = df['age'].map({range(18,24):'age 18-24', range(25,40):'age 25-40', range(41,55):'age 41-55', range(55,100):'age 55+'})