Python 极点：在固定点扩展窗口

这似乎只是每组的常规累积总和:

df.with_columns(
   cum_sum = pl.col("value").cum_sum().over("group")
)

shape: (11, 4)
┌────────────┬───────┬───────┬─────────┐
│ date       ┆ group ┆ value ┆ cum_sum │
│ ---        ┆ ---   ┆ ---   ┆ ---     │
│ date       ┆ i64   ┆ i64   ┆ i64     │
╞════════════╪═══════╪═══════╪═════════╡
│ 2020-03-01 ┆ 1     ┆ 1     ┆ 1       │
│ 2020-05-01 ┆ 1     ┆ 2     ┆ 3       │
│ 2020-11-01 ┆ 1     ┆ 3     ┆ 6       │
│ 2021-01-01 ┆ 1     ┆ 4     ┆ 10      │
│ 2021-02-03 ┆ 1     ┆ 5     ┆ 15      │
│ 2021-06-08 ┆ 1     ┆ 6     ┆ 21      │
│ 2022-01-05 ┆ 1     ┆ 7     ┆ 28      │
│ 2020-07-01 ┆ 2     ┆ 1     ┆ 1       │
│ 2020-09-01 ┆ 2     ┆ 2     ┆ 3       │
│ 2022-01-05 ┆ 2     ┆ 3     ┆ 6       │
│ 2023-02-04 ┆ 2     ┆ 4     ┆ 10      │
└────────────┴───────┴───────┴─────────┘

然后保留每一年的第一行(按组):

df.with_columns(
   cum_sum = pl.col("value").cum_sum().over("group")
).filter(
   pl.col("date").dt.year().is_first_distinct().over("group")
)

shape: (6, 4)
┌────────────┬───────┬───────┬─────────┐
│ date       ┆ group ┆ value ┆ cum_sum │
│ ---        ┆ ---   ┆ ---   ┆ ---     │
│ date       ┆ i64   ┆ i64   ┆ i64     │
╞════════════╪═══════╪═══════╪═════════╡
│ 2020-03-01 ┆ 1     ┆ 1     ┆ 1       │
│ 2021-01-01 ┆ 1     ┆ 4     ┆ 10      │
│ 2022-01-05 ┆ 1     ┆ 7     ┆ 28      │
│ 2020-07-01 ┆ 2     ┆ 1     ┆ 1       │
│ 2022-01-05 ┆ 2     ┆ 3     ┆ 6       │
│ 2023-02-04 ┆ 2     ┆ 4     ┆ 10      │
└────────────┴───────┴───────┴─────────┘

一种更"直接"的方法可能是首先获取年度总和，然后从中生成CUM_SUM.

(df.group_by(
      "group", 
      pl.col("date").dt.year().alias("year"),
      maintain_order=True
   )
   .agg(
      pl.col("date").first(),
      pl.col("value").first(), 
      pl.col("value").sum().alias("sum"),
   )
   .with_columns(
      pl.when(pl.col("group").is_first_distinct())
        .then(pl.col("value"))
        .otherwise(
           pl.col("value") + pl.col("sum").cum_sum().shift().over("group")
        )
        .alias("cum_sum")
   )
)

shape: (6, 6)
┌───────┬──────┬────────────┬───────┬─────┬─────────┐
│ group ┆ year ┆ date       ┆ value ┆ sum ┆ cum_sum │
│ ---   ┆ ---  ┆ ---        ┆ ---   ┆ --- ┆ ---     │
│ i64   ┆ i32  ┆ date       ┆ i64   ┆ i64 ┆ i64     │
╞═══════╪══════╪════════════╪═══════╪═════╪═════════╡
│ 1     ┆ 2020 ┆ 2020-03-01 ┆ 1     ┆ 6   ┆ 1       │
│ 1     ┆ 2021 ┆ 2021-01-01 ┆ 4     ┆ 15  ┆ 10      │
│ 1     ┆ 2022 ┆ 2022-01-05 ┆ 7     ┆ 7   ┆ 28      │
│ 2     ┆ 2020 ┆ 2020-07-01 ┆ 1     ┆ 3   ┆ 1       │
│ 2     ┆ 2022 ┆ 2022-01-05 ┆ 3     ┆ 3   ┆ 6       │
│ 2     ┆ 2023 ┆ 2023-02-04 ┆ 4     ┆ 4   ┆ 10      │
└───────┴──────┴────────────┴───────┴─────┴─────────┘