Как погасить цикл в моем коде при расчете EMWA？ ⇐ Python

[Anonymous]

Я вычисляю значения EWMA для массива потока, и код выглядит следующим образом:

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

import polars as pl
import numpy as np

streamflow_data = np.arange(0, 20, 1)
adaptive_alphas = np.concatenate([np.repeat(0.3, 10), np.repeat(0.6, 10)])
streamflow_series = pl.Series(streamflow_data)
ewma_data = np.zeros_like(streamflow_data)
for i in range(1, len(streamflow_series)):
current_alpha = adaptive_alphas[i]
ewma_data[i] = streamflow_series[:i+1].ewm_mean(alpha=current_alpha)[-1]

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

# I don't round this result
Output: [0  0  1  1  2  3  4  5  6  6  9 10 11 12 13 14 15 16 17 18]

Но если длинаstreamflow_data очень велика (например, >100000), этот код станет очень медленным.
И как же это сделать? можно ли погасить в своем коде и не повлиять на его результат?
Надеюсь на ваш ответ.

Подробнее здесь: https://stackoverflow.com/questions/793 ... ating-emwa