Create age-groups in pandas

X_train_data = pd.DataFrame({'Age':[0,2,4,13,35,-1,54]})

bins= [0,2,4,13,20,110]
labels = ['Infant','Toddler','Kid','Teen','Adult']
X_train_data['AgeGroup'] = pd.cut(X_train_data['Age'], bins=bins, labels=labels, right=False)
print (X_train_data)
   Age AgeGroup
0    0   Infant
1    2  Toddler
2    4      Kid
3   13     Teen
4   35    Adult
5   -1      NaN
6   54    Adult

Source: stackoverflow.com

Similar codes

2 votes176 views1 answer
Django load model by name
Python
2 votes167 views2 answers
Python check if key exist in json
Python
11 votes176 views2 answers
Add pip to path
Python
3 votes177 views1 answer
Python - exclude rowin data frame based on value
Python
1 vote173 views1 answer
Flask enumerate index
Python
1 vote175 views3 answers
Dfs algorithm
Python
1 vote182 views1 answer
How to download the captions of a youtube video
Python
2 votes170 views1 answer
Get all files within multiple directories python
Python
0 vote177 views1 answer
Create a date value array in python
Python
2 votes161 views1 answer
Python ping ip address
Python