Как использовать Pandas скользящее окно с пользовательской функцией, которая требует свернутые данные? - PullRequest
0 голосов
/ 14 апреля 2020

У меня есть функция, которая имеет 3 атрибута: data_string, interval и ratio.

def apx_ent(data_string, interval, ratio):
      return np.mean(data_string) * interval * ratio

Ниже приведены примеры данных, которые у меня есть:

0   300.24
1   312.86
2   302.46
3   297.46
4   274.23
5   288.42
6   274.36
7   248.11
8   269.32
9   239.85
10  252.80
11  240.00
12  240.51
13  228.80
14  222.95
15  243.15

Я бы хотел запустить эту пользовательскую функцию для скользящего окна, однако data_string будет исходить из самого скользящего окна:

df.rolling(3).apply(apx_ent(data_string, 2, 3))

Есть ли у вас какие-либо предложения о том, как выполнить sh это ?

Ожидаемый результат:

0   NAN
1   NAN
2   NAN
3   0.53
4   0.23
5   0.11
6   0.22
...
...