/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

0 vote141 views1 answer
Matplotlib savefig legend cut off
Python
2 votes139 views1 answer
How to reapete the code in python
Python
0 vote137 views1 answer
Get the time of 1 minute later in python
Python
0 vote226 views1 answer
Huggingface default cache dir
Python
16 votes140 views3 answers
How to change the column order in pandas dataframe
Python
0 vote131 views1 answer
Command in python to make negative number positive
Python
0 vote160 views1 answer
Recorrer diccionario python
Python
12 votes138 views3 answers
How to get the url of the current page in selenium python
Python
0 vote127 views1 answer
Mutiple codition datafrarme
Python
6 votes135 views4 answers
What is a python module
Python