Eu tento adicionar uma nova coluna "energy_class" a um dataframe "df_energy" que contenha a string "high" se o valor "consumer_energy" for> 400, "medium" se o valor "consumer_energy" estiver entre 200 e 400 e " baixo "se o valor" consumer_energy "estiver abaixo de 200. Eu tento usar np.where de numpy, mas vejo quenumpy.where(condition[, x, y]) tratar apenas duas condições e não 3 como no meu caso.

Alguma idéia para me ajudar por favor?

Agradeço antecipadamente