/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

0 vote239 views1 answer
Python dlib
Python
11 votes236 views2 answers
Python find and replace string in file
Python
0 vote253 views1 answer
How to find the ascii value of character in python
Python
0 vote247 views1 answer
Split the string using the default arguments
Python
1 vote240 views1 answer
Use django taggit in template
Python Django
0 vote238 views1 answer
Remove watermark using python
Python
3 votes219 views1 answer
Accessing items of tuple in python
Python
59 votes241 views6 answers
How to split word in python
Python
22 votes234 views3 answers
How to call a random function in python
Python
0 vote251 views1 answer
Gensim wordvector vocabulary list
Python