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，我不明白这一点 array [：] thing

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)

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

1 个答案: