Accessing range of values within lists in Numpy

Question

Let's say I have two vectors of size n containing coordinates (point1 and point2), and some large Numpy array with n dimensions (len(array.shape) == 3).

Now, all values of point1 are smaller than point2 and I want to extract the subarray contained between point1 and point2. If I knew the number of dimensions n beforehand (e.g. n=3), I would access it like this:

array[point1[0]:point2[0], point1[1]:point2[1], point1[2]:point2[2]]

I was wondering if there was a clean pythonic way to do this in Numpy that would work for any number of dimensions?

Thanks!

eph · Accepted Answer · 2015-12-02 02:11:49Z

2

array[map(slice,point1,point2)]

The index of A[0:2,0:2] is the same as (slice(0,2), slice(0,2)) which is a tuple of slice.

answered Dec 2, 2015 at 2:11

eph

2,04813 silver badges25 bronze badges

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

Collectives™ on Stack Overflow

Accessing range of values within lists in Numpy

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