Поляры: расширение окна в фиксированных точках ⇐ Python

[Anonymous]

У меня есть

Код: Выделить всё

Polars

dataframe with 3 columns - group, date, value. The goal is to calculate

Код: Выделить всё

cumsum(value)

for each expanding window ends at the first time point at each year for each

Код: Выделить всё

group

.
For example, for the following sample dataframe:

Код: Выделить всё

import polars as pl

df = pl.DataFrame(
{
"date": [
"2020-03-01",
"2020-05-01",
"2020-11-01",
"2021-01-01",
"2021-02-03",
"2021-06-08",
"2022-01-05",
"2020-07-01",
"2020-09-01",
"2022-01-05",
"2023-02-04",
],
"group": [1, 1, 1, 1, 1, 1, 1, 2, 2, 2, 2],
"value": [1, 2, 3, 4, 5, 6, 7, 1, 2, 3, 4],
},
).with_columns(pl.col("date").str.strptime(pl.Date))

The result I am looking for is:

Код: Выделить всё

┌────────────┬───────┬───────┐
│ date       ┆ group ┆ value │
│ ---        ┆ ---   ┆ ---   │
│ date       ┆ i64   ┆ i64   │
╞════════════╪═══════╪═══════╡
│ 2020-03-01 ┆ 1     ┆ 1     │
│ 2021-01-01 ┆ 1     ┆ 10    │
│ 2022-01-05 ┆ 1     ┆ 28    │
│ 2020-07-01 ┆ 2     ┆ 1     │
│ 2022-01-05 ┆ 2     ┆ 6     │
│ 2023-02-04 ┆ 2     ┆ 10    │
└────────────┴───────┴───────┘

Basically, at the first date of each year, calculate the cumulative sum of

Код: Выделить всё

value

from beginning up to (including) this particular date, for each group respectively.
I tried

Код: Выделить всё

group_by_dynamic

and

Код: Выделить всё

rolling

, but still unable to find a concise and clear way to solve this problem.
Any idea is welcome. Thanks!


Источник: https://stackoverflow.com/questions/781 ... xed-points