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 vote116 views1 answer
Convert a text file data to dataframe in python without pandas
Python
0 vote77 views1 answer
HTTP Error 403: Forbidden django account signup email
Python Django
6 votes112 views1 answer
Recursionerror maximum recursion depth
Python
0 vote112 views3 answers
How to print tables using python
Python
1 vote111 views2 answers
Number of columns with no missing values
Python
0 vote121 views1 answer
Position in array python
Python
1 vote119 views1 answer
How to run python file from cmd in dockerfile
Python
2 votes108 views1 answer
How to downgrade python to 3.7 4 anaconda
Python
0 vote109 views1 answer
Is there a way to refer back to a previous line in python
Python
0 vote131 views1 answer
Split a string into N equal parts.
Python