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 vote274 views1 answer
Pandas read chunk of csv
Python
1 vote274 views1 answer
Python list add if not present
Python
12 votes247 views3 answers
Delete all files in a directory python
Python
0 vote287 views1 answer
Print format round python
Python
1 vote279 views1 answer
Finding the Variance and Standard Deviation of a list of numbers in Python
Python
1 vote276 views1 answer
Sin function in python does it values in radians
Python
63 votes285 views3 answers
How to calculate sum of a list in python
Python
108 votes185 views17 answers
Conda environment setup
Python
1 vote267 views2 answers
Isprime function in python
Python
9 votes258 views4 answers
Multiple lines input python
Python