/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 vote332 views1 answer
Selenium.common.exceptions.TimeoutException: Message:
Python
1 vote351 views1 answer
Django wait for database
Python Django
0 vote309 views1 answer
Unable to get local issuer certificate python
Python
0 vote296 views1 answer
Higher-order functions and operations on callable objects in python
Python
0 vote319 views1 answer
How to format a matrix to align all rows python
Python
0 vote320 views1 answer
Pillow rgb to grayscale
Python
0 vote310 views1 answer
Instaed of: newlist = [] for word in wordlist: newlist.append(word.upper())
Python
0 vote347 views1 answer
Sklearn model persistence
Python
0 vote313 views1 answer
Py2exe no console
Python
0 vote308 views1 answer
Python submatrix
Python