Как использовать библиотеку Polars в Python для постепенной пометки последовательных единиц? ⇐ Python

[Anonymous]

Как мне пометить последовательные появления единиц в наборе данных с помощью Polars? Я нашел здесь ответ на очень похожий вопрос. Я пытаюсь пометить данные уникальным целочисленным идентификатором, чтобы позже выполнить группировку или поворот. Я буду использовать те же данные, что и в вопросе, на который есть ссылка.

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

import polars as pl

data = {'Switch': [0, 1, 1, 0, 1, 1, 1, 0, 0, 1, 0, 1, 1, 1, 1, 0, 1, 0, 0, 1, 1, 0, 0, 0, 1, 1, 1, 1, 0, 1, 0, 0, 1]}
df = pl.DataFrame(data)

Вывод должен быть

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

shape: (33, 2)
┌────────┬─────────────────────┐
│ Switch ┆ Switch Event Number │
│ ---    ┆ ---                 │
│ i64    ┆ u32                 │
╞════════╪═════════════════════╡
│ 0      ┆ 0                   │
│ 1      ┆ 1                   │
│ 1      ┆ 1                   │
│ 0      ┆ 0                   │
│ 1      ┆ 2                   │
│ 1      ┆ 2                   │
│ 1      ┆ 2                   │
│ 0      ┆ 0                   │
│ 0      ┆ 0                   │
│ 1      ┆ 3                   │
│ 0      ┆ 0                   │
│ 1      ┆ 4                   │
│ 1      ┆ 4                   │
│ 1      ┆ 4                   │
│ 1      ┆ 4                   │
│ 0      ┆ 0                   │
│ 1      ┆ 5                   │
│ 0      ┆ 0                   │
│ 0      ┆ 0                   │
│ 1      ┆ 6                   │
│ 1      ┆ 6                   │
│ 0      ┆ 0                   │
│ 0      ┆ 0                   │
│ 0      ┆ 0                   │
│ 1      ┆ 7                   │
│ 1      ┆ 7                   │
│ 1      ┆ 7                   │
│ 1      ┆ 7                   │
│ 0      ┆ 0                   │
│ 1      ┆ 8                   │
│ 0      ┆ 0                   │
│ 0      ┆ 0                   │
│ 1      ┆ 9                   │
└────────┴─────────────────────┘

Я новичок в Polars и практически запутался.

Подробнее здесь: https://stackoverflow.com/questions/767 ... ecutive-1s