rebuild numpy array, need faster method

Question

I have image numpy array (640,480,3) where X,Y is coordinates and (255 255 255) is a color mask for point

I try to get new 2d array(x,y), where X and Y is coordinates for point when color > zero..

I try this code and it works, but it takes too much processor time

enter code here
for x in range(edges.shape[0]):
     for y in range(edges.shape[1]):
          if edges[[x],[y],[0]]!=0:
             new.append([x,y])

Divakar · Accepted Answer · 2016-11-03 22:30:54Z

2

You could slice the first element of last axis, compare it against 0 and then use np.argwhere to get those indices, which would be the x, y coordinates in a (N,2) shaped array.

Thus, the implementation would be simply -

new = np.argwhere( edges[...,0]!=0 )

answered Nov 3, 2016 at 22:30

Divakar

222k19 gold badges273 silver badges374 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

rebuild numpy array, need faster method

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related