Select a subset from 2D numpy array

Question

From a 2D numpy array Z, I'd like to remove certain points, at indicies ix, iy.
ix, iy are currently sequential lists of the x-index and corresponding y-index, like so:

ix = range(10)
iy = range(10,20,1)

So I'd like Z[0][10], Z[1][11], etc. put into a new list or array.

What is the best (read: most pythonic) way of doing this? Perhaps there is a better way of finding these indicies so I get an array of [ix,iy], [ix+1,iy+1] etc.?

I have looked at this question. I don't know how to use the list comprehension syntax of the selected answer for 2 dimensions.

roman · Accepted Answer · 2013-08-06 10:44:50Z

2

Can't test now, but AFAIK it could be done like Z[ix, iy] in numpy.
Also, you don't have to write range(10,20,1), you can use range(10,20)

answered Aug 6, 2013 at 10:44

roman

118k30 gold badges205 silver badges209 bronze badges

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

Comments

lynn · Accepted Answer · 2013-08-06 10:38:52Z

0

[Z[y][x] for x, y in zip(ix, iy)]

(What zip(ix, iy) does is: make a list of tuples [(ix[0], iy[0]), (ix[1], iy[1]), ...], i.e. the x and y indices for each element we need.)

answered Aug 6, 2013 at 10:38

lynn

10.9k1 gold badge48 silver badges81 bronze badges

1 Comment

ehacinom Over a year ago

This works too, but I had to flip the index to Z[x][y], which depends on how you define the x-axis. Thanks!

Collectives™ on Stack Overflow

Select a subset from 2D numpy array

2 Answers 2

Comments

1 Comment

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

1 Comment

Your Answer

Sign up or log in

Post as a guest

Linked

Related