python - 跨数组求和模式

Question

我很难找到正确的方法来做一些我认为使用 numpy 应该很简单的事情。我有一个数组 (1000x1000)，我想计算数组中特定模式的总和。

例如:

如果我有这个数组，并且想要计算两个单元格右对角线的总和，我希望 [7,12,11,8,12,6,11,7] (共8笔)。

我该怎么做？

Answer 1

此操作称为二维 convolution :

>>> import numpy as np
>>> from scipy.signal import convolve2d
>>> kernel = np.eye(2, dtype=int)
>>> a = np.array([[5,3,7,1,2],[3,2,9,4,7],[8,9,4,2,3]])
>>> convolve2d(a, kernel, mode='valid')
array([[ 7, 12, 11,  8],
       [12,  6, 11,  7]])

如果您想将其推广到任意维度，还有 scipy.ndimage.convolve可用的。它也适用于这种 2d 情况，但不提供 mode='valid' 便利。