python - 在 python 中计算二维网格上的一组的有效方法是什么？ [下图]

Question

我有几百个 2D numpy 数组。它们包含零和一。一些带有绘图的示例，黄色表示 1，紫色表示 0:

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

plt.imshow(grid1)

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

plt.imshow(grid2)

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

plt.imshow(grid3)

我正在寻找一种有效的方法来计算图像上黄色 Blob 的数量。上图中的 2、1 和 4 个 Blob ，从上到下。

有没有一种简单的方法可以做到这一点，或者我必须检查每个黄色位是否与所有其他黄色位在同一个 blob 中，并为此编写一个脚本？ (那看起来很痛苦。)

Answer 1

scipy.ndimage.measurements.label 满足您的需求。