Роллинг Ewma в пандах

Question

Я пытаюсь вычислить ewma по мере поступления, точно так же, как скользящее среднее, std и т. Д. Код, приведенный ниже, не работает, я не уверен, в чем проблема.

import pandas as pd
import numpy as np


a = pd.DataFrame(np.random.randn(50, 5))
pd.rolling_apply(arg=a[0], 
                 func=lambda x: pd.ewma(x, span=10), 
                 window=10)

Ошибка типа: в скаляры Python могут быть преобразованы только массивы размера 1

Answer 1

rolling_apply устарела в пандах 0.23 (текущая версия)

Для работы с EWMA используйте DataFrame.ewm

a.ewm(span=10, adjust=False).mean()