/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 vote185 views1 answer
Insert a new row to numpy array in especific position
Python
0 vote133 views1 answer
No module named 'django_hitcounter'
Python
0 vote138 views1 answer
Write a python program to find table of a number using while loop
Python
1 vote129 views1 answer
Python tkinter getting labels
Python
1 vote139 views1 answer
Python catch sigterm
Python
93 votes136 views4 answers
How to install python pyautogui
Python
0 vote129 views1 answer
Poython inl linrt dor loop
Python
22 votes140 views4 answers
How to write string in python
Python
0 vote125 views1 answer
Reset Python Dictionary to 0 Zero. Empty existing Python Dictionary
Python
0 vote117 views1 answer
Join mulitple dataframe pandas index
Python