python - 在滑动窗口中求和元素 - NumPy

Question

有一种numpy方法可以对区间中的每三个元素求和吗？例如:

import numpy as np
mydata = np.array([4, 2, 3, 8, -6, 10])

我想得到这个结果:

np.array([9, 13, 5, 12])

Answer 1

我们可以使用np.convolve -

np.convolve(mydata,np.ones(3,dtype=int),'valid')

convolution 的基本思想是我们有一个内核，我们在输入数组中滑动，卷积运算在内核滑动时将元素乘以内核元素相加。因此，为了解决窗口大小为 3 的情况，我们使用由 np.ones(3) 生成的三个 1 内核.

sample 运行-

In [334]: mydata
Out[334]: array([ 4,  2,  3,  8, -6, 10])

In [335]: np.convolve(mydata,np.ones(3,dtype=int),'valid')
Out[335]: array([ 9, 13,  5, 12])