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 votes316 views3 answers
Fibonacci series python using function
Python
1 vote309 views1 answer
Set icon title tkinter
Python
0 vote316 views1 answer
Python branchless programming
Python
0 vote347 views1 answer
Making a txt file then write
Python
0 vote309 views1 answer
Elongated muskrat
Python
0 vote301 views1 answer
Listview
Python
0 vote315 views1 answer
A way to extract from a DateTime value data without seconds
Python
2 votes315 views2 answers
Convert np shape (a,) to (a,1)
Python
3 votes277 views2 answers
Upgrade python to 3.8
Python
-1 vote290 views1 answer
Save a file as a pickle
Python