NumpyScipy 连通组件

Question

我正在用python编写一个程序，在L*L矩阵中寻找由1、0或-1组成的“孤岛”。 我需要它来找到这些连接组件的“区域”，标记每个元素，并且能够返回矩阵m[x][y]的给定元素所属岛屿的大小。在

import numpy as np
from scipy import ndimage
from scipy import misc
import matplotlib.pyplot as plt

m = np.random.randint(-1,2,(L,L))

mask1 = (m == -1)
mask2 = (m == 0)
mask3 = (m == 1)

label_m1, nb_labels1 = ndimage.label(mask1)
label_m2, nb_labels2 = ndimage.label(mask2)
label_m3, nb_labels3 = ndimage.label(mask3)

这应该给我一个随机矩阵m的标记岛（只是一个例子），但我不知道如何得到一个给定点所属的“区域”的大小。 你能帮帮我吗？我几乎没有编程经验，所以请原谅我，如果问题是愚蠢的。在

谢谢

Answer 1

比如：

label_count_m1 = np.bincount(label_m1.ravel())
label_count_m1[0] = 0
sizes_m1 = label_count_m1[label_m1]