/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

-1 vote385 views2 answers
Python increment
Python
1 vote375 views1 answer
Db_index django
Python Django
4 votes316 views5 answers
Python read xml
Python
0 vote366 views1 answer
Check if value in dictionary keys python dataframe
Python
11 votes332 views1 answer
UnicodeDecodeError: 'utf-8' codec can't decode byte 0xe2 in position 10: invalid continuation byte
Python
0 vote357 views1 answer
Id
Python
0 vote350 views1 answer
Create contract from interface in brownie
Python
4 votes338 views1 answer
Python django shell command
Python
2 votes344 views1 answer
What does * mean in python in functions
Python
1 vote376 views1 answer
How to get the user ip in djagno
Python