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 vote147 views1 answer
Python how to turn a word into a list
Python
1 vote163 views1 answer
Maximum and minimum value of array python
Python
12 votes160 views6 answers
Python check if nan
Python
0 vote147 views1 answer
Django rest framework not getting form
Python
0 vote166 views1 answer
Bulk upload with dictionary or list in django moels
Python
1 vote192 views1 answer
How to give multiple option to the user and ask the same question again and again until the user tells one of the options
Python
1 vote147 views1 answer
Python validate url
Python
0 vote145 views1 answer
Pysft connection drop issue
Python
0 vote184 views10 answers
Sqlite3 with flask web application CRUD pdf
Python
0 vote173 views1 answer
Strftime python multiple formats
Python