Группировать по + скользящее среднее в Polars ⇐ Python

[Anonymous]

Мне нужно перевести этот фрагмент кода pandas:

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

df.groupby(groupby_col)[[A_col, B_col]].rolling(window=window, on=B_col, closed = 'left').agg(some_function)

к полярам.
A_col имеет Int64 dtype, B_col datetime (ns).
Я придумал следующую формулировку:

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

df.group_by(groupby_col).agg(pl.col([A_col, B_col])).with_columns(
pl.col([A_col, B_col]).rolling_mean(window_size=30,  by=B_col, closed="left"))

что привело меня к этой ошибке:

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

`expr_name` operation not supported for dtype `list[date]` (expected: date/datetime)

Я не могу понять, почему он не может использовать список дат в качестве скользящего среднего значения «по».

Подробнее здесь: https://stackoverflow.com/questions/778 ... -in-polars