/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

10 votes519 views2 answers
Sum of a column in pandas
Python
0 vote213 views1 answer
Python set list index value that doesn't exist
Python
0 vote219 views1 answer
How to download from youtube in discord.py
Python
0 vote210 views1 answer
Specific mail.search python UNSEEN SINCE T
Python
0 vote213 views1 answer
Save jupyter notebook session
Python
9 votes210 views1 answer
Cv2 grayscale
Python
0 vote193 views1 answer
Python crear variables
Python
1 vote218 views1 answer
Strptime python decimal seconds
Python
3 votes162 views1 answer
Centos v8.11.1 npm gyp err! stack error: can't find python executable "python", you can set the python env variable.
Python
0 vote208 views1 answer
Verilog import module
Python