Python Polars库中的滚动索引

您可以从给出每个组的最大计数(在聚合内使用pl.len())的方法开始，并对Counter列进行后处理，以使其值在每个组内增加.

(
    df
    .rolling(index_column="Date", period="2d", group_by="Name", check_sorted=False)
    .agg(
        pl.len().alias("Counter")
    )
    .with_columns(
        (pl.col("Counter") - pl.len() + 1 + pl.int_range(pl.len())).over("Name", "Date")
    )
)

shape: (11, 3)
┌─────────┬────────────┬─────────┐
│ Name    ┆ Date       ┆ Counter │
│ ---     ┆ ---        ┆ ---     │
│ str     ┆ date       ┆ i64     │
╞═════════╪════════════╪═════════╡
│ John    ┆ 2023-01-01 ┆ 1       │
│ John    ┆ 2023-01-01 ┆ 2       │
│ John    ┆ 2023-01-01 ┆ 3       │
│ John    ┆ 2023-01-01 ┆ 4       │
│ John    ┆ 2023-01-02 ┆ 5       │
│ John    ┆ 2023-01-02 ┆ 6       │
│ John    ┆ 2023-01-02 ┆ 7       │
│ John    ┆ 2023-01-02 ┆ 8       │
│ John    ┆ 2023-01-03 ┆ 5       │
│ John    ┆ 2023-01-03 ┆ 6       │
│ New Guy ┆ 2023-01-01 ┆ 1       │
└─────────┴────────────┴─────────┘

Explanation.聚合后，Counter列将在每个长度为L的名称-日期-组内取常数值V.目标是让Counter取V-L+1到V的值(每行一个值).

因此我们可以

1. Counter减go L-1，
2. 添加一个值从0到L-1增加的int范围.