/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 vote104 views1 answer
Python Couldn't find a tree builder with the features you requested: lxml. Do you need to install a parser library?
Python
0 vote125 views1 answer
Python define random variables name
Python
7 votes138 views2 answers
How to code python
Python
0 vote112 views1 answer
@property python
Python
8 votes127 views2 answers
Determinant of matrix in python
Python
0 vote110 views1 answer
Filter dict by list of keys python
Python
0 vote96 views1 answer
For in list start with index python
Python
4 votes111 views5 answers
Module 'datetime' has no attribute 'now' django
Python
0 vote107 views1 answer
Python as-lookup
Python
0 vote111 views2 answers
Leer fichero de texto con columnas como diccionario python
Python