Используйте функцию прокрутки поляров, чтобы получить список всех значений в прокручивающихся окнах. ⇐ Python

[Anonymous]

Я хотел бы использовать функцию прокрутки, чтобы получить список всех значений в прокручивающихся окнах.
Я попробовал это с помощью следующего фрагмента кода:

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

import polars as pl

df = pl.DataFrame(
{
"A": [1, 2, 9, 2, 13],
}
)

df.select(
pl.col("A").rolling_map(lambda s: s, 3),
)

это выводит

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

┌──────┐
│ A    │
│ ---  │
│ i64  │
╞══════╡
│ null │
│ null │
│ 1    │
│ 2    │
│ 9    │
└──────┘

но мне нужно:

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

shape: (5, 1)
┌─────────────────┐
│ A               │
│ ---             │
│ list[i64]       │
╞═════════════════╡
│ [null, null, 1] │
│ [null, 1, 2]    │
│ [1, 2, 9]       │
│ [2, 9, 2]       │
│ [9, 2, 13]      │
└─────────────────┘

Есть ли у кого-нибудь идеи, как это сделать в полярах простым способом?

Подробнее здесь: https://stackoverflow.com/questions/703 ... rolling-wi