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

2 votes320 views1 answer
Write custom query odoo
Python
69 votes202 views6 answers
How to activate venv python
Python
10 votes209 views2 answers
Replace values in a column by condition python
Python
26 votes219 views2 answers
Python time library
Python
0 vote289 views1 answer
TypeError: 'LazyCorpusLoader' object is not callable
Python
0 vote203 views1 answer
Change x axis frequency
Python
0 vote212 views1 answer
Change size of image and fir it into numpy array opencv
Python
0 vote211 views1 answer
Selenium session id python
Python
69 votes201 views6 answers
Python dictionary append
Python
-1 vote205 views1 answer
Python - how many letters are capital in a string
Python