/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

5 votes209 views1 answer
Pynput.keyboard.Key
Python
1 vote322 views1 answer
Backup django db from one database to another
Python Django
0 vote189 views1 answer
Python - subtracting dictionary values
Python
34 votes230 views6 answers
Dataframe from dict
Python
0 vote234 views1 answer
Django serialize a get object is not iterable
Python Django
2 votes207 views1 answer
Pandas transform
Python
133 votes215 views6 answers
Input function in python
Python
4 votes225 views2 answers
How to make django model field case insensitive
Python
10 votes246 views5 answers
Python with file
Python
1 vote230 views1 answer
Who wrote permission to dance
Python