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

[Anonymous]

В настоящее время я пытаюсь создать новый столбец в кадре данных Polars (df). В моем df много строк, и в этом новом столбце я хочу, чтобы мой существующий список заполнялся только там, где выполняются определенные условия.
import polars as pl

df = pl.DataFrame({
'cond1': [-1, 0, 1, 2, 2, 3, 4, 5],
'cond2': ['No', 'No', 'No', 'Yes','Yes', 'Yes', 'Yes', 'Yes']
})

Я знаю, что в pandas я мог бы сделать что-то вроде следующего (где «выход» — это мой новый столбец и предполагается, что список имеет ту же длину, что и отфильтрованный df):
df_pd = df.to_pandas()

conditions = ((df_pd['cond1']>=1) & (df_pd['cond2'] == 'Yes'))
list_new = [1,2,3,4,5]

df_pd.loc[conditions, 'output'] = list_new

# cond1 cond2 output
# 0 -1 No NaN
# 1 0 No NaN
# 2 1 No NaN
# 3 2 Yes 1.0
# 4 2 Yes 2.0
# 5 3 Yes 3.0
# 6 4 Yes 4.0
# 7 5 Yes 5.0

Как бы я сделал то же самое, но в пределах поляров? Я хочу, чтобы общее количество строк оставалось прежним, я просто хочу обновить новый столбец везде, где выполняются условия.
Я попробовал несколько вариантов следующих действий, но ничего не помогло:
df = df.with_columns(pl.when(conditions).then(list_new).alias('output')


Подробнее здесь: https://stackoverflow.com/questions/786 ... rows-match