python - 如何获取二进制掩码的位置真实位

Question

我知道我可以通过简单的循环获得真实的位，但是循环速度慢得离谱

   locations = []
   width, height = mask.shape[0], mask.shape[1]
   for x in range(len(0,width)):
      for y in range(len(0,height):
        if mask[x][y] is 1:
          location.append([x,y])

我想知道如何避免循环开销。我正在学习python，但我不明白这个数组[:] 东西

Answer 1

您可以使用非零。它返回按轴分组的索引 - 换句话说，形式为 (array([x1, x2, x3,...]), array([y1, y2, y3,...] ), 数组([z1, z2, z3,...]),...):

>>> a = numpy.array([[0, 0], [0, 1], [1, 0]], dtype=numpy.bool)
>>> numpy.nonzero(a)
(array([1, 2]), array([1, 0]))

您可以使用结果作为索引来获取非零值:

>>> a[numpy.nonzero(a)]
array([ True,  True], dtype=bool)