如何使用具有需要滚动数据的自定义功能的Pandas滚动窗口?

时间:2020-04-14 15:26:45

标签: python pandas rolling-computation

我有一个具有3个属性的函数:data_stringintervalratio

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))

您对如何完成此操作有任何建议吗?

预期输出:

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

0 个答案:

没有答案