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

Любая идея, чтобы помочь мне, пожалуйста?

заранее спасибо