/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

36 votes235 views4 answers
Python print string in red color
Python
0 vote221 views1 answer
Ec2 upgrade python 3.7 to 3.8
Python
0 vote229 views1 answer
Change period to timestamp python
Python
-1 vote194 views1 answer
Django any() function raw sql query
Python
3 votes241 views2 answers
Uses specific version python venv
Python
0 vote233 views1 answer
Python reading into a text file and diplaying items in a user friendly manner
Python
2 votes247 views1 answer
Python pandas apply to one column
Python
3 votes236 views2 answers
Python script that executes at time
Python
0 vote239 views1 answer
Install python 3.6 ubuntu 16.04
Python
-1 vote247 views1 answer
Set form field disabled django
Python Django