Скользящий подсчет полярностей с временным окном ⇐ Python

[Anonymous]

Я пытаюсь написать метод для конвейера функций, который возвращает полярное выражение. Метод должен принимать имя столбца в виде строки и целое число дней. Я хочу выполнить скользящий подсчет в этом столбце, используя окно, равное количеству дней.
Похоже, что выраженияrolling_count не существует, поэтому я попытался использоватьrolling_sum_by, но безрезультатно.

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

def temporal_rolling_count(col: str, days: int) -> pl.Expr:
return (
pl.lit(1)
.rolling_sum_by(window_size=f"{days}d", by="date_time")
.over(col)
.fill_null(0)
)

Я также пробовал этот метод, который был ближе, но все же не во всех случаях работал

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

def temporal_rolling_count(col: str, days: int) -> pl.Expr:
return (
pl.col(col)
.cum_count()
.over(col, (pl.col("date_time") - pl.col("date_time").min()).dt.days() % days == 0)
.fill_null(0)
)

Можно ли как-нибудь добиться этого, вернув выражение? Или мне придется действовать непосредственно с DataFrame, возможно, с помощью прокрутки?

Подробнее здесь: https://stackoverflow.com/questions/758 ... ral-window