Np where pandas with 3 choices

col         = 'consumption_energy'
conditions  = [ df2[col] >= 400, (df2[col] < 400) & (df2[col]> 200), df2[col] <= 200 ]
choices     = [ "high", 'medium', 'low' ]
    
df2["energy_class"] = np.select(conditions, choices, default=np.nan)


  consumption_energy energy_class
0                 459         high
1                 416         high
2                 186          low
3                 250       medium
4                 411         high
5                 210       medium
6                 343       medium
7                 328       medium
8                 208       medium
9                 223       medium

Source: stackoverflow.com

Similar codes

-1 vote202 views1 answer
Change tuple in python
Python
1 vote197 views1 answer
Img read
Python
0 vote201 views1 answer
Csrf token fetch django
Python Django
1 vote192 views1 answer
Draw line from 2 mouse event in image python
Python
0 vote195 views1 answer
Natsort python pip install
Python
1 vote240 views1 answer
Module 'torch' has no 'stack' memberpylint(no-member)
Python
3 votes164 views2 answers
Django favicon
Python Django
0 vote246 views1 answer
Matplotlib FiveThirtyEight creating a signature
Python
1 vote171 views2 answers
Add new row to numpy array
Python
39 votes206 views2 answers
String to list of characters python
Python