/Python/Add age categories pandas dataframe

Add age categories pandas dataframe

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

22 votes319 views7 answers
Remove special characters from string python
Python
4 votes432 views1 answer
Django is null
Python Django
29 votes354 views1 answer
Df dropna ensure that one column is not nan
Python
0 vote350 views1 answer
How to add previous and next in tkinter in python
Python
0 vote345 views1 answer
Write to file in p
Python
1 vote340 views1 answer
Python string format_map
Python
-1 vote341 views1 answer
Templates python
Python
22 votes380 views3 answers
Python sort list in reverse order
Python
2 votes335 views1 answer
All alphanumeric characters for python python
Python
0 vote368 views1 answer
How to extract dictionary value from string in python
Python