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 vote271 views2 answers
Python delete duplicate lines in file
Python
120 votes269 views6 answers
Getting Input in Python
Python
0 vote260 views1 answer
Split a string by comma in python
Python
0 vote288 views1 answer
Python range of array
Python
0 vote268 views1 answer
Xlabel not showing matplotlib
Python
0 vote280 views1 answer
Plt clor image histogram
Python
0 vote251 views1 answer
Take first 10 row while reading csv python
Python
0 vote260 views1 answer
Python on_mouse_down/collidepoint
Python
0 vote274 views1 answer
Df filter out rows that appear more than x times
Python
0 vote227 views2 answers
Tkinter disable resize
Python