是否可以在 Polars 中“重置”cummax？

Question

当达到某个值时，我想重新开始累积最大值。例如。每次 cummax() 超过基础值 5% 时。我怎样才能在极地中做到这一点？

Answer 1

假设您想在值超过值 6 时重置 cummax。你可以这样做：

(
    df.with_columns(
        (pl.col("a") >= 6)
        .shift(1)
        .fill_null(False)
        .cumsum()
        .alias("group")
    ).with_columns(
        pl.col("a")
        .cummax()
        .over(pl.col("group"))
        .alias("cummax")
    )
)

例子：

In [73]: df = pl.DataFrame({'a': [1, 3, 5, 6, 1, 4, 3, 6, 5, 6]})

In [74]: (
    ...:     df.with_columns(
    ...:         (pl.col("a") >= 6)
    ...:         .shift(1)
    ...:         .fill_null(False)
    ...:         .cumsum()
    ...:         .alias("group")
    ...:     ).with_columns(
    ...:         pl.col("a")
    ...:         .cummax()
    ...:         .over(pl.col("group"))
    ...:         .alias("cummax")
    ...:     )
    ...: )
Out[74]:
shape: (10, 3)
┌─────┬───────┬────────┐
│ a   ┆ group ┆ cummax │
│ --- ┆ ---   ┆ ---    │
│ i64 ┆ u32   ┆ i64    │
╞═════╪═══════╪════════╡
│ 1   ┆ 0     ┆ 1      │
│ 3   ┆ 0     ┆ 3      │
│ 5   ┆ 0     ┆ 5      │
│ 6   ┆ 0     ┆ 6      │
│ ... ┆ ...   ┆ ...    │
│ 3   ┆ 1     ┆ 4      │
│ 6   ┆ 1     ┆ 6      │
│ 5   ┆ 2     ┆ 5      │
│ 6   ┆ 2     ┆ 6      │
└─────┴───────┴────────┘