Numpy multi-dimensional array slicing

Question

I have a 3-D NumPy array with shape (100, 50, 20). I was trying to slice the third dimension of the array by using the index, e.g., from 1 to 6 and from 8 to 10.

I tried the following code, but it kept reporting a syntax error.

newarr [:,:,1:10] = oldarr[:,:,[1:7,8:11]]

[1:7,8:11] is an invalid use of slices. The error was caught by the interpreter, but it also has no meaning to numpy. That's a discontinuous set of locations, which has to be enumerated as shown in the answer. — hpaulj
– hpaulj, Commented Oct 2, 2021 at 23:01

akuiper · Accepted Answer · 2021-10-02 22:06:46Z

1

You can use np.r_ to concatenate slice objects:

newarr [:,:,1:10] = oldarr[:,:,np.r_[1:7,8:11]]

Example:

np.r_[1:4,6:8]
array([1, 2, 3, 6, 7])

answered Oct 2, 2021 at 22:06

akuiper

216k33 gold badges362 silver badges379 bronze badges

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

Collectives™ on Stack Overflow

Numpy multi-dimensional array slicing

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