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 vote184 views1 answer
Number of unique pairs in columns pandas
Python
0 vote184 views3 answers
How to draw dendrogram in python
Python
0 vote189 views1 answer
Matlab index last element
Python
16 votes213 views6 answers
List to Dataframe
Python
0 vote237 views1 answer
Invalid Python Environment: Python is unable to find Maya's Python modules
Python
0 vote210 views1 answer
How to delete all elements of a list in python
Python
6 votes185 views2 answers
Pandas count rows in column
Python
2 votes245 views3 answers
Run flask in debug mode
Python Flask
0 vote194 views1 answer
Define a function in python without arguments
Python
0 vote186 views1 answer
Python [remote rejected] master -> master (pre-receive hook declined)
Python