Python/Numpy: How to extract interior of any dimension of numpy array?

Question

Suppose I have a numpy array A which can be of any dimensions len(A.shape) can be 1,2,3,..etc. and a corresponding array, crop which len(crop) = len(A.shape) and I want to extract the interior values of A using crop. Here is an example for 2D array.

    A = np.random.rand(30).reshape([5,6])
    crop = np.array([1,2])

Wanted output:

    A[crop[0]:-crop[0], crop[1]:-crop[1])

Assuming value of crop will be reasonable with respect to size of A. How do I do this for any dimension of array A ?

Divakar · Accepted Answer · 2019-09-20 09:51:00Z

4

Here's one way with slice notation -

A[tuple([slice(i,-i,None) for i in crop])]

Or with the shorthand np.s_ -

A[tuple([np.s_[i:-i] for i in crop])]

If the start and end indices are given for each dimension, we can do something like as shown in Slicing NumPy array given start and end indices for generic dimensions.

edited Sep 20, 2019 at 9:51

answered Sep 20, 2019 at 9:18

Divakar

222k19 gold badges273 silver badges374 bronze badges

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

Collectives™ on Stack Overflow

Python/Numpy: How to extract interior of any dimension of numpy 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