Скользящие расчеты KPI в полярах, индекс не отображается

Похоже, вам не нужно группировать, а вместо этого запустить rolling_median() над окном.

• over(), чтобы ограничить вычисления в пределах group_identifier.
• name.suffix() для присвоения имен новым столбцам.

Если вам нужны только отфильтрованные строки:

(
    df
    .filter(mask_for_filter)
    .with_columns(
        pl.col("col_1", "col_2")
        .rolling_median(15, min_periods=1, center=True)
        .over("group_identifier")
        .name.suffix("_median")
    )
)

┌──────────────────┬───────┬───────┬───────────┬──────────────┬──────────────┐
│ group_identifier ┆ col_1 ┆ col_2 ┆ other_col ┆ col_1_median ┆ col_2_median │
│ ---              ┆ ---   ┆ ---   ┆ ---       ┆ ---          ┆ ---          │
│ str              ┆ f64   ┆ f64   ┆ f64       ┆ f64          ┆ f64          │
╞══════════════════╪═══════╪═══════╪═══════════╪══════════════╪══════════════╡
│ B                ┆ 0.01  ┆ 1.68  ┆ 1.12      ┆ 0.83         ┆ -0.46        │
│ B                ┆ 0.37  ┆ -0.26 ┆ 0.04      ┆ 0.85         ┆ -0.66        │
│ A                ┆ 0.72  ┆ -0.38 ┆ 0.47      ┆ 0.93         ┆ -0.44        │
│ A                ┆ 0.36  ┆ -0.51 ┆ -0.4      ┆ 0.86         ┆ -0.5         │
...
└──────────────────┴───────┴───────┴───────────┴──────────────┴──────────────┘

Или, если вам это нужно в исходном DataFrame

• When/then() дважды — верхний, чтобы назначить скользящую медиану только тем строкам, которые имеют col_1 > 0, и второй, чтобы не включать строки, которые будут фильтроваться при вычислении скользящей медианы.

(
    df
    .with_columns(
        pl.when(pl.col("col_1") > 0).then(
            pl.when(pl.col("col_1") > 0).then(pl.col("col_1", "col_2"))
            .rolling_median(15, min_periods=1, center=True)
            .over("group_identifier")
        )
        .name.suffix("_median")
    )
)

Если вы хотите добавить больше агрегатов, вы можете обобщить их (хотя я не уверен, что они достаточно читаемы, чтобы перейти к производству):

(
    df
    .with_columns(
        pl.when(pl.col("col_1") > 0).then(
            transform(
                pl.when(pl.col("col_1") > 0).then(pl.col("col_1", "col_2")),
                15, min_periods=1, center=True
            )
            .over("group_identifier")
        )
        .name.suffix(suffix)
        for transform, suffix in [
            (pl.Expr.rolling_median, "_median"),
            (pl.Expr.rolling_mean, "_mean"),
        ]
    )
)