Python，Numpy。 在二維數組中查找值並用 1 替換鄰居

Question

我有一個 0 和 1 的 10x10 數組。

我想：

1. 找到每個單元格的 position，其值為 1。

2. 用1替換所有鄰居。鄰居=任何單元格到an=1距離（也是對角線）。 例子：

    array([[0, 1, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 1, 0, 0], [1, 0, 0, 0, 0], [0, 0, 0, 1, 1]])

output：

array([[1, 1, 1, 1, 0],
       [1, 1, 1, 1, 0],
       [1, 1, 1, 1, 0],
       [1, 1, 1, 1, 1],
       [1, 1, 1, 1, 1]])

我正在嘗試查找索引，但它不起作用：

a=np.where(a==1)+1

從其他帖子中，我還嘗試使用此 function 獲取鄰居：

def n_closest(x,n,d=1):
    return x[n[0]-d:n[0]+d+1,n[1]-d:n[1]+d+1] 

但這不適用於邊緣

謝謝

Answer 1

如果您不介意使用scipy ，二維卷積將很快解決問題：

import numpy as np
from scipy import signal

# Input array
X = np.array([[0, 1, 0, 0, 0],
              [0, 0, 0, 0, 0],
              [0, 0, 1, 0, 0],
              [1, 0, 0, 0, 0],
              [0, 0, 0, 1, 1]])

# We apply a 2D convolution with a 3x3 kernel and we check which value are bigger than 0.
R = (signal.convolve2d(X,np.ones((3,3)),mode='same')>0).astype(int)
# R = array([[1, 1, 1, 0, 0],
#            [1, 1, 1, 1, 0],
#            [1, 1, 1, 1, 0],
#            [1, 1, 1, 1, 1],
#            [1, 1, 1, 1, 1]])  

# Finally we extract the index 
x,y = np.where(R)