Заполнение нескольких столбцов одновременно с помощью Map_elements поляров [дубликат] ⇐ Python

[Anonymous]

Отвечая на этот вопрос, могу ли я использовать map_elements от Polars для одновременного заполнения нескольких столбцов.
Скажем, у меня есть собственная функция Python

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

def my_complicated_function(row):

# ...

return value1, value2, value3

и я хотел бы использовать это следующим образом

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

df = df.with_columns(
pl.struct(pl.all())
.map_elements(my_complicated_function, return_dtype=pl.Int64)
.alias("column1", "column2", "column3")
)

за исключением того, что псевдоним не создает несколько столбцов одновременно. Могу ли я использовать здесь структуру для вывода или есть что-то еще?

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