Numpy «где» с несколькими условиями ⇐ Python

[Anonymous]

Я пытаюсь добавить новый столбец «energy_class» в фрейм данных «df_energy», который содержит строку «high», если значение «consumption_energy» > 400, «medium», если значение «consumption_energy» находится между 200 и 400, и «низкий», если значение «consumption_energy» меньше 200.
Я пытаюсь использовать np.where из numpy, но вижу, что numpy.where(condition[, x, y]) обрабатывайте только два условия, а не 3, как в моем случае.

Подробнее здесь: https://stackoverflow.com/questions/391 ... conditions