Как я могу получить местоположение истинных биты двоичной маски

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]) 

Я Kidda, что избежать накладных петель. Я изучаю 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)