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

您可以使用nonzero. 它返回按轴分组的索引——换句话说，一个元组的形式(array([x1, x2, x3,...]), array([y1, y2, y3,...]), array([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)