Как вычислить столбец в Polars DataFrame с использованием np.linspace ⇐ Python

[Anonymous]

Рассмотрим следующее pl.dataframe :

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

df = pl.DataFrame(
data={
"np_linspace_start": [0, 0, 0],
"np_linspace_stop": [8, 6, 7],
"np_linspace_num": [5, 4, 4]
}
)

shape: (3, 3)
┌───────────────────┬──────────────────┬─────────────────┐
│ np_linspace_start ┆ np_linspace_stop ┆ np_linspace_num │
│ ---               ┆ ---              ┆ ---             │
│ i64               ┆ i64              ┆ i64             │
╞═══════════════════╪══════════════════╪═════════════════╡
│ 0                 ┆ 8                ┆ 5               │
│ 0                 ┆ 6                ┆ 4               │
│ 0                 ┆ 7                ┆ 4               │
└───────────────────┴──────────────────┴─────────────────┘

Как я могу создать новый столбец ls , то есть результат функции np.linspace ? В этом столбце будет храниться np.array .
Я что -то искал в этом направлении:

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

df.with_columns(
ls=np.linspace(
start=pl.col("np_linspace_start"),
stop=pl.col("np_linspace_stop"),
num=pl.col("np_linspace_num")
)
)

Есть ли поляры эквивалент np.linspace ?

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