Идиоматический способ Polars агрегирования n последовательных строк кадра данных ⇐ Python

[Anonymous]

Я новичок в Polars, и в итоге я написал этот код для вычисления некоторого агрегатного выражения по сегментам из n строк:

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

import polars as pl

df = pl.DataFrame({"a": [1, 1, 3, 8, 62, 535, 4213]})

(
df.with_columns(index=pl.int_range(pl.len(), dtype=pl.Int32))
.group_by_dynamic(index_column="index", every="3i")
.agg(pl.col("a").mean())
)

Для примера я установил n==3 для 7 строк, но подумайте о небольшом n, около 100, для многостолбцового фрейма данных примерно из 10**6 строк.
Мне интересно, является ли это идиоматическим способом выполнения операций такого типа.
Как-то group_by_dynamic над Диапазон Int32 кажется мне излишним: мне было интересно, есть ли более прямой способ выполнить ту же агрегацию.

Подробнее здесь: https://stackoverflow.com/questions/779 ... data-frame