如何区分图形或数组中的多个形状？

Question

我在 Python 中有一个二维数组，其中包含 0 或 1 的值，排列成各种形状。对于我当前的项目，我需要一种方法来区分图像中的形状。

我目前正在尝试通过将第一个形状的值设置为 1，将第二个形状的值设置为 2 等等来做到这一点。然而，事实证明这对我来说太难编码了，所以我想知道是否有人知道一个好的算法来做到这一点，或者知道另一种方法。谢谢。

Answer 1

我怀疑您正在寻找label features，它可以在scipy.ndimage 中找到。尽管名称如此，但它不一定特定于图像分析。引用文档“输入中的任何非零值都算作特征，零值被视为背景。”

结果将是一个数组，大小相同，特征编号。例如以下代码：

import numpy as np
from scipy.ndimage import measurements

arr = np.array([
[0,0,0,0,0,0],
[0,1,1,0,0,0],
[0,1,1,0,0,1],
[0,0,0,0,1,1],
[0,0,0,0,1,1]
])

labeled_array, number_of_features = measurements.label(arr)
print(labeled_array)

...将产生以下输出，两个特征分别编号为 1 和 2：

array([[0, 0, 0, 0, 0, 0],
       [0, 1, 1, 0, 0, 0],
       [0, 1, 1, 0, 0, 2],
       [0, 0, 0, 0, 2, 2],
       [0, 0, 0, 0, 2, 2]], dtype=int32)

第二个返回参数包含数组中的number_of_features（这里是2）。