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

0 vote181 views1 answer
Tf.slice
Python
0 vote112 views2 answers
Merge two sorted lists into one sorted list
Python
0 vote63 views1 answer
Print g
Python
2 votes110 views1 answer
Change size of selenium window
Python
2 votes112 views1 answer
Title() in python
Python
1 vote98 views1 answer
Python how to draw triangle
Python
0 vote105 views1 answer
How to vonvert 1 d list to 2d list in pytohn
Python
4 votes97 views2 answers
One hot encoding python pandas
Python
0 vote101 views1 answer
Python find cells with na
Python
0 vote108 views1 answer
Python as-lookup
Python