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 vote313 views2 answers
Sorted set in python
Python
9 votes305 views2 answers
Python script to read all file names in a folder
Python
14 votes315 views4 answers
Python split every character in string
Python
0 vote318 views1 answer
Plt.hist using bins
Python
0 vote304 views1 answer
Nlp = spacy.load('en') error
Python
0 vote331 views1 answer
Use get method request html page python
Python
-2 votes319 views1 answer
Numpy difference between two arrays
Python
3 votes319 views1 answer
How to change window size in kivy python
Python
0 vote344 views1 answer
Django cms api
Python
4 votes319 views1 answer
How to add column headers in pandas
Python