Группировка строк в пакеты в Polars с помощью скользящего окна ⇐ Python

[Anonymous]

Я пытаюсь выяснить, как объединить строки в пакеты (т. е. скользящее/переворачивающееся окно), чтобы у нас была таблица типа

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

  id   x    y
---- ---- ----
1   x1   y1
2   x2   y2
3   x3   y3
4   x4   y4
5   x5   y5

Результат будет

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

  id   x    y                  Grouped x3
---- ---- ---- -----------------------------------------
1   x1   y1   None
2   x2   y2   None
3   x3   y3   [[1, x1, y1], [2, x2, y2], [3, x3, y3]]
4   x4   y4   [[2, x2, y2], [3, x3, y3], [4, x4, y4]]
5   x5   y5   [[3, x3, y3], [4, x4, y4], [5, x5, y5]]

В настоящее время у меня есть агрегированный столбец с использованием concat_list, такой как:

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

  id   x    y       List
---- ---- ---- -------------
1   x1   y1   [1, x1, y1]
2   x2   y2   [2, x2, y2]
3   x3   y3   [3, x3, y3]
4   x4   y4   [4, x4, y4]
5   x5   y5   [5, x5, y5]

Я просто не знаю, как агрегировать их по строкам, как в примере выше.
Большое спасибо!

Подробнее здесь: https://stackoverflow.com/questions/725 ... ing-window