/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

4 votes213 views2 answers
Django reset database
Python Django
12 votes183 views4 answers
Dictionary append value python
Python
15 votes219 views4 answers
Q django
Python Django
2 votes194 views5 answers
Pandas rename single column
Python
0 vote199 views4 answers
Pip django graphql
Python Django
1 vote188 views1 answer
Python string math
Python
0 vote185 views1 answer
Set .discard(), .remove() & .pop() | Hackerrank Python Solution |
Python
0 vote211 views1 answer
How to install pandas for aws sam local
Python
13 votes179 views5 answers
How to create role discord.py
Python
9 votes216 views1 answer
Bee movie script
Python