python移动窗口函数

在Python中，可以使用numpy库中的rolling函数来实现移动窗口函数。

该函数的语法为：

numpy.rolling(window, axis=None)[source]

其中，window参数表示窗口大小，axis参数表示在哪个轴上进行移动窗口运算。默认为None，表示在所有轴上进行运算。

例如，下面的代码演示了如何对一维数组进行移动窗口均值运算：

import numpy as np

data = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
window_size = 3

moving_avg = np.convolve(data, np.ones(window_size)/window_size, mode='valid')

print(moving_avg)

运行结果为：

[ 2.  3.  4.  5.  6.  7.  8.  9.]

在上面的代码中，首先定义了一个一维数组data和窗口大小window_size。然后，使用np.convolve函数对data进行卷积运算，卷积核为一个大小为window_size的全1数组除以窗口大小window_size。最后，使用mode参数设置为'valid'，表示输出大小与原始数组大小相同，即不进行填充操作。