Как найти группы последовательных единиц в Polars? ⇐ Python

[Anonymous]

У меня есть следующий DataFrame:

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

import polars as pl

pl.Config(tbl_rows=16)

data = [0, 1, 1, 0, 1, 1, 1, 0, 0, 1, 0, 1, 1, 1, 1, 0]

df = pl.DataFrame({"test": data})

Я хочу добиться следующего результата:

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

┌──────┬────────┐
│ test ┆ result │
│ ---  ┆ ---    │
│ i64  ┆ u32    │
╞══════╪════════╡
│ 0    ┆ 0      │
│ 1    ┆ 1      │
│ 1    ┆ 2      │
│ 0    ┆ 0      │
│ 1    ┆ 1      │
│ 1    ┆ 2      │
│ 1    ┆ 3      │
│ 0    ┆ 0      │
│ 0    ┆ 0      │
│ 1    ┆ 1      │
│ 0    ┆ 0      │
│ 1    ┆ 1      │
│ 1    ┆ 2      │
│ 1    ┆ 3      │
│ 1    ┆ 4      │
│ 0    ┆ 0      │
└──────┴────────┘

Желаемый результат — сохранить исходные значения 0 неизменными, начать накапливать последовательные 1 и сбросить счетчик до начального значения при обнаружении значения 0.
Объем данных настолько велик, что синтаксис, аналогичный циклу for, использовать невозможно.
Что делать, если мне необходимо использовать только функции Polars, а не numpy или другие библиотеки?>

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