/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 vote282 views1 answer
Add data to foreign key table django
Python Django
27 votes295 views4 answers
Create dictionary python from two lists
Python
12 votes301 views5 answers
Turtle write
Python
0 vote291 views4 answers
Mergesort
Python
0 vote354 views1 answer
Flask - store object directly in a session [duplicate]
Python Flask
-2 votes335 views1 answer
Type python fast
Python
0 vote279 views1 answer
Getting multiple of 5 python
Python
0 vote312 views2 answers
Pandas fillna multiple columns
Python
0 vote273 views1 answer
Discord.py - change the default help command
Python
0 vote298 views1 answer
Pandas correlation matrix between one column and all others
Python