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

[Anonymous]

Я хотел бы сгруппировать значения так, чтобы разница между каждым элементом группы оставалась в пределах определенного процента. Например. каждый раз, когда элемент на 5% превышает первый элемент группы, он переходит в новую группу. В качестве возврата мне нужно первое значение группы. Пример с порогом 5 %, где задано значение "a", необходимо рассчитать значения "group" и "groupFirst":
import polars as pl

df = pl.DataFrame({'a': [100, 103, 105, 106, 105, 104, 103, 106, 100, 102],
'group': [0, 0, 1, 1, 1, 1, 1, 1, 2, 2],
'groupFirst': [100, 100, 105, 105, 105, 105, 105, 105, 100, 100]})

print(df)

shape: (10, 3)
┌─────┬───────┬────────────┐
│ a ┆ group ┆ groupFirst │
│ --- ┆ --- ┆ --- │
│ i64 ┆ i64 ┆ i64 │
╞═════╪═══════╪════════════╡
│ 100 ┆ 0 ┆ 100 │
│ 103 ┆ 0 ┆ 100 │
│ 105 ┆ 1 ┆ 105 │
│ 106 ┆ 1 ┆ 105 │
│ 105 ┆ 1 ┆ 105 │
│ 104 ┆ 1 ┆ 105 │
│ 103 ┆ 1 ┆ 105 │
│ 106 ┆ 1 ┆ 105 │
│ 100 ┆ 2 ┆ 100 │
│ 102 ┆ 2 ┆ 100 │
└─────┴───────┴────────────┘


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