python - 二维数组中 8 个邻居的总和

Question

我需要找到一个单元格的所有相邻元素的总和，例如 getsumofneighbors(matrix, i, j):

'M*N matrix'
[[0 1 0]
 [2 0 1]
 [0 4 0]
 [0 0 0]]

单元格 [0][0] 的最近元素之和为 3

[1][0] 是 5

在 [1][1] 处是 8

是否有一个 python 库可以找到给定单元格旁边所有元素的总和？

Answer 1

如果不介意对scipy的依赖，可以使用scipy.ndimage.convolve ，如下:

In [475]: a
Out[475]: 
array([[0, 1, 0],
       [2, 0, 1],
       [0, 4, 0],
       [0, 0, 0]])

In [476]: kernel
Out[476]: 
array([[1, 1, 1],
       [1, 0, 1],
       [1, 1, 1]])

In [477]: from scipy.ndimage import convolve

In [478]: c = convolve(a, kernel, mode='constant')

In [479]: c
Out[479]: 
array([[3, 3, 2],
       [5, 8, 5],
       [6, 3, 5],
       [4, 4, 4]])