Как добавить новый столбец в фрейм данных Polars, значение которого зависит от разных условий ⇐ Python

[Anonymous]

У меня есть фрейм данных, содержащий несколько полей, и я хотел бы добавить дополнительный столбец в качестве метки на основе этих полей. В идеале код должен выглядеть примерно так:

Код: Выделить всё

df.with_columns(
[
pl.when(~pl.all(pl.col("temp").is_null()) & pl.all(pl.col("daily_temp").is_null())).then("sensor_1_" + pl.col("location")).alias("label"),
pl.when(pl.all(pl.col("temp").is_null()) & ~pl.all(pl.col("daily_temp").is_null())).then("sensor_2_" + pl.col("loc")).alias("label"),
]
)

Условия не имеют значения, однако я хотел бы иметь возможность использовать несколько условий, как описано выше, для добавления значения в один и тот же столбец. Использование описанного кода приведет к возникновению ошибки дублирования столбца. Есть ли способ сделать это правильно?

Подробнее здесь: https://stackoverflow.com/questions/768 ... -on-differ