Group_by и добавьте столбец счетчика в кадр данных Polars [дубликат] ⇐ Python

[Anonymous]

У меня есть фрейм данных Polars
import polars as pl

df = pl.from_repr("""
┌─────────┬─────────────────────────────┐
│ item_id ┆ num_days_after_first_review │
│ --- ┆ --- │
│ i64 ┆ i64 │
╞═════════╪═════════════════════════════╡
│ 1 ┆ 3 │
│ 1 ┆ 3 │
│ 1 ┆ 10 │
│ 2 ┆ 2 │
│ 2 ┆ 2 │
│ 3 ┆ 1 │
│ 3 ┆ 5 │
└─────────┴─────────────────────────────┘
""")

Мне хотелось бы иметь столбец, в котором указан счетчик для каждого item_id относительно num_days_after_first_review;
чтобы результат был таким:
shape: (7, 3)
┌─────────┬─────────────────────────────┬─────┐
│ item_id ┆ num_days_after_first_review ┆ num │
│ --- ┆ --- ┆ --- │
│ i64 ┆ i64 ┆ i64 │
╞═════════╪═════════════════════════════╪═════╡
│ 1 ┆ 3 ┆ 1 │
│ 1 ┆ 3 ┆ 2 │
│ 1 ┆ 10 ┆ 3 │
│ 2 ┆ 1 ┆ 1 │
│ 2 ┆ 2 ┆ 2 │
│ 3 ┆ 1 ┆ 1 │
│ 3 ┆ 5 ┆ 2 │
└─────────┴─────────────────────────────┴─────┘


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