/Python/Equivalent of case_when in r in pandas

Equivalent of case_when in r in pandas

conditions = [
    (df["age"].lt(10)),
    (df["age"].ge(10) & df["age"].lt(20)),
    (df["age"].ge(20) & df["age"].lt(30)),
    (df["age"].ge(30) & df["age"].lt(50)),
    (df["age"].ge(50)),
]
choices = ["baby", "kid", "young", "mature", "grandpa"]

df["elderly"] = np.select(conditions, choices)

# Results in:
#      name  age  preTestScore  postTestScore  elderly
#  0  Jason   42             4             25   mature
#  1  Molly   52            24             94  grandpa
#  2   Tina   36            31             57   mature
#  3   Jake   24             2             62    young
#  4    Amy   73             3             70  grandpa

Source: stackoverflow.com

Similar codes

2 votes275 views3 answers
How to open file explorer in python
Python
0 vote259 views1 answer
Drop row with duplicate value
Python
19 votes266 views1 answer
Python cut string to length
Python
0 vote263 views1 answer
Python run scp command
Python
1 vote277 views1 answer
Python: separate lines including the period or excalamtion mark and print it to the prompt..
Python
1 vote267 views1 answer
What is print in python
Python
0 vote325 views1 answer
Manage.py startapp not working in django
Python Django
53 votes289 views2 answers
Add values to dictionary key python
Python
0 vote257 views1 answer
How to see what variable is closest to a higher element in python
Python
0 vote241 views7 answers
Convert python to java online
Python