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

80 votes203 views4 answers
Pip for python
Python
15 votes210 views5 answers
Join on column pandas
Python
0 vote214 views1 answer
Ascii to decimal python
Python
4 votes201 views2 answers
Number length python
Python
2 votes208 views2 answers
Python program to find first n prime numbers
Python
0 vote231 views5 answers
Medium seaaborn mathplot diesign styles
Python
0 vote235 views1 answer
Create fixtures django
Python
42 votes201 views6 answers
Python regex
Python
33 votes193 views4 answers
How to install pyaudio
Python
0 vote221 views1 answer
Gnome-terminal stopped executing after python 3.6 is installed
Python