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

12 votes314 views3 answers
Python for in for in
Python
3 votes313 views1 answer
Django import model from another app
Python
-1 vote305 views1 answer
Show all text contained in a line pandaq
Python
15 votes353 views2 answers
Post request in python flaks
Python
0 vote319 views1 answer
Phone no validate
Python
0 vote338 views1 answer
Convert a pdf folder to excell pandas
Python
0 vote391 views1 answer
TypeError: __init__(): incompatible constructor arguments. The following argument types are supported: 1. tensorflow.python._pywrap_file_io.BufferedInputStream(arg0: str, arg1: int)
Python
-2 votes325 views4 answers
Merge multiple dataframs
Python
1 vote385 views1 answer
How to add an active class to current element in navbar in django
Python Django
0 vote363 views1 answer
Identify the common columns between two dataframes pandas python
Python