Посмотрите вперед в group_by_dynamic ⇐ Python

[Anonymous]

При выполнении group_by_dynamic для столбцов индекса и агрегировании столбцов в списке (чтобы увидеть, какие значения находятся в группе) для данной группы и данного индекса i значения в списке соответствуют значению в правильной группе, но для индекса i, i+1, ... до периода.
Этот просмотр вперед в индексе, кажется, контрастирует с функциейrolling_mean, которая (по умолчанию) выполняет прокрутку значений более низких индексов, а не более высоких. означает.
Это намеренно разработано таким образом, и если да, то как можно выполнить group_by_dynamic с использованием нижних индексов? (я не уверен, что делает параметр offset, но это не то, что я хочу здесь)
Вот пример, который я ожидал не поднимать

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

import polars as pl

df = pl.DataFrame(
{
"index": [0, 0, 1, 1],
"group": ["banana", "pear", "banana", "pear"],
"weight": [2, 3, 5, 7],
}
)

agg = df.group_by_dynamic("index", group_by="group", every="1i", period="2i").agg(pl.col("weight"))

assert((
agg
.filter(index=0, group="banana")
.select("weight")
.to_series()
.to_list()
) == [[2]])

Спасибо


Подробнее здесь: https://stackoverflow.com/questions/758 ... by-dynamic