Как группировать_по и скользящую сумму в полярах？ ⇐ Python

[Anonymous]

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

import polars as pl

data = {'type': ['A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'B', 'B', 'B', 'B', 'B', 'B', 'B', 'B', 'B', 'C', 'C', 'C', 'C', 'C', 'C', 'C', 'C', 'C', 'C', 'C'],
'value': [5, 9, 1, 0, 3, 2, 5, 8, 9, 1, 0, 3, 3, 1, 1, 0, 2, 0, 0, 5, 7, 4, 7, 8, 9, 11, 1, 1, 0, 1, 4, 3, 21]}
df = pl.DataFrame(data)
print(df)

Имея два столбца данных, как мы можем сгруппировать их по столбцу «тип», суммировать столбец «значение», используя скользящее окно размером 5, а затем поместить полученные данные в столбец с именем «результат»?
Результаты следующие:

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

[None, None, None, None, 18, 15, 11, 18, 27, 25, 23, 21, 16, None, None, None, None, 4, 3, 7, 14, 16, None, None, None, None, 36, 30, 22, 14, 7, 9, 29]

(Используйте только библиотеку Polars, версия Polars = 0.17.9)

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