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

1 vote408 views1 answer
How get 1st column in all rows of a 2d matrix in python
Python
0 vote256 views1 answer
Gcsfs pd.read_parquet
Python
1 vote255 views1 answer
The name tf.summary.merge_all is deprecated. Please use tf.compat.v1.summary.merge_all
Python
0 vote262 views1 answer
Negative slicing in python list
Python
19 votes281 views5 answers
Python floor
Python
1 vote281 views1 answer
Python strftime microseconds
Python
5 votes383 views3 answers
ImportError: No module named flask
Python Flask
1 vote279 views1 answer
Plt normalized histogram
Python
0 vote311 views1 answer
Django not detecting new model
Python Django
10 votes289 views2 answers
Update anaconda from cmd
Python