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

0 vote338 views1 answer
Python string strip non alphanumeric
Python
48 votes349 views3 answers
Online python compiler with modules
Python
1 vote357 views1 answer
Selenium api
Python
0 vote342 views1 answer
How to make a operating system in python
Python
1 vote413 views1 answer
Pygame disable message
Python
0 vote340 views1 answer
Start loop at 1 python
Python
0 vote341 views2 answers
AttributeError: module 'keras.utils' has no attribute 'to_categorical'
Python
0 vote314 views1 answer
Pandas if value present in df index
Python
0 vote435 views1 answer
How to compress image field in django?
Python
0 vote334 views1 answer
From decimal to binary python without using any libraries
Python