python - Numpy:查找至少有一个零相邻的元素的最有效方法 ("edges")

Question

假设我有一个像这样的 numpy 数组:

np.array([[0, 0, 0, 0, 0],
          [0, 1, 0, 0, 0],
          [1, 1, 1, 0, 0],
          [2, 2, 2, 1, 0],
          [3, 3, 3, 2, 1]])

如何获得与它相邻的零正交值的索引？
通过正交我的意思是它不会查看对角相邻的项目。
对于示例，输出应为:

np.array([[1, 1], [2, 0], [2, 2], [3, 3], [4, 4]])

Answer 1

明显的方法有什么问题吗？

mask = arr == 0
out = np.zeros(mask.shape, dtype='bool')

out[:-1] = out[:-1] | mask[1:]
out[1:] = out[1:] | mask[:-1]
out[:,:-1] = out[:,:-1] | mask[:,1:]
out[:,1:] = out[:,1:] | mask[:,:-1]

np.argwhere(out & (arr!=0))