/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

16 votes254 views6 answers
Dataframe to dictionary
Python
55 votes247 views5 answers
How to clear the console python
Python
0 vote239 views1 answer
Support vector machine svm using python numerical example
Python
0 vote267 views1 answer
Stop a subprocess python
Python
0 vote309 views1 answer
Python print f
Python
1 vote243 views1 answer
USB: usb_device_handle_win.cc:1049 Failed to read descriptor from node connection: A device attached to the system is not functioning. (0x1F)
Python
1 vote291 views1 answer
How do i check if a django queryset is empty
Python Django
0 vote207 views1 answer
Function that return a lenght of list
Python
0 vote233 views2 answers
How to convert c to python
Python
1 vote235 views2 answers
Python line break inside string
Python