Объединение нескольких столбцов в список в одном столбце ⇐ Python

[Anonymous]

Я хотел бы объединить несколько столбцов в виде списка в один столбец.
Например, этот фрейм данных:

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

import polars as pl
import numpy as np

df = pl.from_repr("""
┌─────┬─────┐
│ a   ┆ b   │
│ --- ┆ --- │
│ i64 ┆ i64 │
╞═════╪═════╡
│ 1   ┆ 4   │
│ 2   ┆ 5   │
│ 3   ┆ 6   │
└─────┴─────┘
""")

в этот:

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

┌────────────┐
│ combine    │
│ ---        │
│ list [i64] │
╞════════════╡
│ [1, 4]     │
│ [2, 5]     │
│ [3, 6]     │
└────────────┘

Сейчас я делаю это так:

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

df = df.with_columns(pl.map_batches(['a','b'],lambda df:pl.Series(np.column_stack([df[0].to_numpy(),df[1].to_numpy()]).tolist())).alias('combine'))

Есть ли лучший способ сделать это?


Подробнее здесь: https://stackoverflow.com/questions/701 ... gle-column