slice index using numpy array-CodePudding

Consider the following code:

import numpy as np
np.random.seed(0)

x = np.random.randint(0,9,(10,10,3))
x[np.array([0,1]):np.array([5,6])]

I get the following error:

TypeError: only integer scalar arrays can be converted to a scalar index

I guess I would need to unpack the arrays that index x but can't figure out a way to do it since * doesn't work.

The goal is to have x[0:5] and x[1:6]

CodePudding user response：

You could generate an index array similar to this answer.

import numpy as np

x = np.random.randint(0,9,(10,10,3))
idcs = np.array([
    np.arange(0,5),
    np.arange(1,6),
])

print(x.shape)
# (10, 10, 3)
print(x[idcs].shape)
# (2, 5, 10, 3)

CodePudding user response：

if I have understood the question right , you should not use ":" in x[np.array([0,1]):np.array([0,1])]

use "," instead x[np.array([0,1]),np.array([0,1])]

CodePudding user response：

here is the another one,

import numpy as np
np.random.seed(0)

x = np.random.randint(0,9,(10,10,3))
[x[i:j] for i,j in zip(np.array([0,1]),np.array([5,6]))]