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

2 votes181 views1 answer
Pip install on different version of python
Python
24 votes485 views6 answers
Static files in django
Python Django
2 votes225 views1 answer
Check corently installed epython version
Python
1 vote241 views1 answer
Python close gile
Python
13 votes158 views2 answers
Pd.set_option('display.max_columns' none)
Python
0 vote204 views1 answer
Add new dimension
Python
3 votes201 views1 answer
Boto3 upload file to s3
Python
0 vote189 views1 answer
Distance between numpy arrays
Python
0 vote176 views1 answer
Spark.read.load
Python
4 votes207 views1 answer
Django forms set class
Python