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

57 votes196 views3 answers
How to add an item to a dictionary in python
Python
0 vote236 views1 answer
How to search for a data in excel pandas
Python
8 votes224 views4 answers
Set index to column pandas
Python
6 votes230 views5 answers
No module named env.__main__; 'env' is a package and cannot be directly executed
Python
10 votes204 views1 answer
Enumerate zip python
Python
0 vote206 views1 answer
Pandas swapaxes example
Python
0 vote218 views1 answer
How to check if string is in byte formate pythin
Python
1 vote198 views1 answer
File to lowercase python
Python
0 vote210 views1 answer
Improt kmean
Python
0 vote220 views1 answer
How to filter a series in pandas
Python