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 vote198 views1 answer
Python pair two lists into a dictionary
Python
0 vote179 views1 answer
Calculate pearson's r | calculate graph correlation
Python
0 vote158 views1 answer
Proj 4.9.0 must be installed.
Python
1 vote173 views2 answers
Sns figsize
Python
0 vote224 views1 answer
How to deploy to heroku djangorest framework app
Python Django
0 vote166 views1 answer
Python 10-e10 numbers
Python
-1 vote176 views1 answer
Harihar kaka class 10 questions
Python
0 vote160 views1 answer
Download textdocuments with python
Python
0 vote180 views1 answer
Tk inter entry
Python
3 votes171 views3 answers
Pandas merge two dataframes remove duplicates
Python