填充形状的内部

Question

我正在处理一个问题，我处理了很多形状，例如：

​

​

它们被表示为简单的numpy数组，0表示空，任何其他值表示非空。

我需要填充内部空间为空的形状，例如上面的两个正方形。但我还没有想出任何简单的解决方案。

对包有什么想法或建议吗？

Answer 1

您可以使用scipy.ndimage.binary_fill_holes来实现此目的：

import scipy

a = np.array([
    [0, 0, 1, 1, 1],
    [0, 0, 1, 0, 1],
    [0, 0, 1, 1, 1],
    [0, 0, 0, 0, 0],
    [2, 2, 2, 0, 0],
    [2, 0, 2, 0, 0],
    [2, 2, 2, 0, 0],
])
b = scipy.ndimage.binary_fill_holes(a)
print(b.astype(int))
# [[0 0 1 1 1]
#  [0 0 1 1 1]
#  [0 0 1 1 1]
#  [0 0 0 0 0]
#  [1 1 1 0 0]
#  [1 1 1 0 0]
#  [1 1 1 0 0]]

这样做的一个问题是它不能保持形状id。如果你想这样，可能会更复杂一点……如果你没有一个形状在另一个形状里面的情况，也许这是可行的：

import scipy

a = np.array([
    [0, 0, 1, 1, 1],
    [0, 0, 1, 0, 1],
    [0, 0, 1, 1, 1],
    [0, 0, 0, 0, 0],
    [2, 2, 2, 0, 0],
    [2, 0, 2, 0, 0],
    [2, 2, 2, 0, 0],
])
b = np.zeros_like(a)
for i in range(a.max()):
    shape = i + 1
    b += shape * scipy.ndimage.binary_fill_holes(a == shape)
print(b)
# [[0 0 1 1 1]
#  [0 0 1 1 1]
#  [0 0 1 1 1]
#  [0 0 0 0 0]
#  [2 2 2 0 0]
#  [2 2 2 0 0]
#  [2 2 2 0 0]]

仍然有一些复杂的案例。例如，输入：

1 1 1
1 0 2
2 2 2

将在第一个代码段中生成所有1，而在第二个代码段中不做任何更改。根据您的确切需求，您可以根据需要调整代码。