Question

所以我有一个2D Numpy数组，看起来像这样：

[[1,1,1,2,2],
 [1,1,1,2,2],
 [1,2,2,2,2]]

其中数组中的每个数字代表一个区域。我想生成一个布尔数组，该数组在相邻元素不相等的位置上显示 True ，即在区域边界上。所以结果应该是这样的：

[[False, False, True, False],
...etc

我知道一个简单的双for循环可以做到这一点，但我需要更快的东西。

Answer 1

一般来说，您正在寻找边缘检测滤波器。

有多种方法可以解决这个问题，但基本的想法是将一个简单的过滤器（例如[-1, 1]或[-1, 0, 1]）与您的数据进行卷积。 scipy.ndimage和scikit-image是开始这个的好地方。请记住，处理一般情况的方法不止一种。

但是，如果您想要最简单的情况，可以使用基本的numpy操作：

import numpy as np

x = np.array([[1,1,1,2,2],
              [1,1,1,2,2],
              [1,2,2,2,2]])

edges = np.abs(np.diff(x, axis=1)) > 0

这会产生：

array([[False, False,  True, False],
       [False, False,  True, False],
       [ True, False, False, False]], dtype=bool)

如果您希望输出具有与输入相同的形状，则可以使用几个不同的选项。您可以通过左侧或右侧边缘值填充输入数组：

# Pad the right-hand side with the edge values
# Use [(0, 0), (1, 0)] to pad the left instead.
xpad = np.pad(x, [(0, 0), (0, 1)], mode='edge')

edges = np.abs(np.diff(xpad, axis=1)) > 0

哪会产生：

array([[False, False,  True, False, False],
       [False, False,  True, False, False],
       [ True, False, False, False, False]], dtype=bool)

在这种特定情况下，您可以使用np.gradient而不是填充数组。 gradient将在边缘使用不同的算法来确保维护数组大小：

dy, dx = np.gradient(x)
edges = np.abs(dx) > 0

这会产生稍厚的边框，因为它使用的算法与简单地减去相邻元素的算法略有不同：

array([[False, False,  True,  True, False],
       [False, False,  True,  True, False],
       [ True,  True, False, False, False]], dtype=bool)

最后，根据“双边框”的含义，您可能会尝试使用比简单单边差异更复杂的边缘滤镜。例如：

In [4]: import scipy.ndimage

In [5]: abs(scipy.ndimage.laplace(x)) > 0
Out[5]:
array([[False, False,  True,  True, False],
       [False,  True,  True,  True, False],
       [ True,  True,  True, False, False]], dtype=bool)

在2D numpy网格中查找相邻元素

1 个答案: