Как в Polars обновить несколько столбцов одновременно? ⇐ Python

[Anonymous]

Предположим, у нас есть примерно такая рамка Polars

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

lf = pl.LazyFrame([
pl.Series("a", ...),
pl.Series("b", ...),
pl.Series("c", ...),
pl.Series("i", ...)
])

и такая функция

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

def update(a, b, c, i):
s = a + b + c + i
a /= s
b /= s
c /= s
return a, b, c

это зависит от элементов столбцов a, b, c, а также i.
Как мы можем обновить каждый строку кадра с помощью функции?
Мы могли бы использовать with_columns для независимого обновления строк каждого столбца, но как мы можем сделать это, учитывая зависимость между столбцами?

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