/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

2 votes152 views1 answer
Django setup allowed hosts
Python
3 votes160 views2 answers
Import class from another file python
Python
0 vote146 views1 answer
Change order of barh matplotlib
Python
0 vote145 views1 answer
Return object in python
Python
0 vote145 views1 answer
Pd assign index from different df
Python
4 votes180 views1 answer
Video streaming flask
Python Flask
11 votes153 views3 answers
Python set windows environment variable
Python
1 vote157 views1 answer
Calculate mode in python
Python
0 vote170 views1 answer
Installing intel python
Python
27 votes156 views4 answers
Create a dataframe from dict
Python