/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

3 votes312 views3 answers
Flask migrate install
Python Flask
0 vote273 views3 answers
Python run subprocess and get output
Python
0 vote270 views1 answer
Extract nonzero array elements python
Python
9 votes318 views3 answers
Takes 2 positional arguments but 3 were given
Python
2 votes282 views2 answers
How to add elements in a dictionary in python
Python
0 vote290 views1 answer
B'[AUTHENTICATIONFAILED] Invalid credentials (Failure)'
Python
1 vote346 views1 answer
Procfile flask
Python Flask
8 votes269 views4 answers
ImportError: No module named pandas
Python
4 votes287 views1 answer
Python check if value in string
Python
24 votes279 views5 answers
Get time python
Python