polars.Series.rolling_map

Series.rolling_map(
    function: Callable[[Series],
    Any],
    window_size: int,
    weights: list_[float] | None = None,
    *,
    min_samples: int | None = None,
    center: bool = False,
) → Series

Вычисляет пользовательскую функцию для скользящего окна.

Предупреждение

Эта функциональность считается нестабильной. Она может быть изменена в любой момент, и это не будет считаться нарушением обратной совместимости.

Изменено в версии 1.21.0: Параметр min_periods переименован в min_samples.

Параметры:

function

Пользовательская функция агрегации.

window_size

Длина окна в количестве элементов.

weights

Необязательный срез той же длины, что и окно, который будет поэлементно умножен на значения в окне.

min_samples

Количество значений в окне, которые должны быть не null, прежде чем будет вычислен результат. Если задано None (по умолчанию), оно будет равно window_size.

center

Размещает метки в центре окна.

Предупреждение

Вычисление пользовательских функций выполняется крайне медленно. По возможности используйте специализированные функции скользящего окна, такие как Series.rolling_sum().

Примеры

>>> from numpy import nansum
>>> s = pl.Series([11.0, 2.0, 9.0, float("nan"), 8.0])
>>> s.rolling_map(nansum, window_size=3)
shape: (5,)
Series: '' [f64]
[
    null
    null
    22.0
    11.0
    17.0
]