index image with 2D array

Question

I have an image of size (1080,1920,3) stored in a numpy array and am trying to index it using an index array of size (6,2). My code is below and it runs, however the output dimensions are (2,6,3), which is not what I expect.

A = inPix[np.arange(6), 2darray]

This method above is from the answers here. Can anyone help me understand whats wrong and how I can get a (6,3) output array?

chris · Accepted Answer · 2016-12-15 08:52:44Z

0

I found out that if you split the indices into columns, it works.

A = inPix[2darray[:,0],2darray[:,1]]

answered Dec 15, 2016 at 8:52

chris

5,0665 gold badges40 silver badges69 bronze badges

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

Collectives™ on Stack Overflow

index image with 2D array

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related