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

26 votes300 views4 answers
Python global variables
Python
0 vote233 views1 answer
Change variable type python
Python
3 votes247 views3 answers
Python - find specific name in a df
Python
3 votes239 views1 answer
Python not equal to
Python
2 votes238 views1 answer
Spyder - comment banch of codee
Python
1 vote249 views1 answer
Django making a custom 403 page
Python
1 vote224 views1 answer
Cartesian product of a list python
Python
0 vote287 views1 answer
Difference between __gte and __lte in Django
Python Django
15 votes233 views3 answers
Discord.py send image
Python
26 votes259 views6 answers
Sort list of dictionaries python
Python