python - numpy中X[:, :, :, i]是什么意思？

Question

我在 Python 中有这个子代码，我无法理解它是什么或它做什么，尤其是这个语句:

X[:,:,:,i]

子代码是:

train_dict = sio.loadmat(train_location)
X = np.asarray(train_dict['X'])

X_train = []
for i in range(X.shape[3]):
    X_train.append(X[:,:,:,i])
X_train = np.asarray(X_train)

Y_train = train_dict['y']
for i in range(len(Y_train)):
    if Y_train[i]%10 == 0:
        Y_train[i] = 0
Y_train = to_categorical(Y_train,10)
return (X_train,Y_train)

Answer 1

这称为数组切片。正如@cᴏʟᴅsᴘᴇᴇᴅ 提到的，x是一个 4D 数组和 X[:,:,:,i]获取它的一个特定 3D 数组切片。

也许一个维度较少的例子会有所帮助。

matrix = np.arange(4).reshape((2,2))

在这种情况下 matrix是一个二维数组:

array([[0, 1],
       [2, 3]])

因此 matrix[:, 1]将导致 matrix 的较小切片:

array([1, 3])

原代码 matrix[:,:,:, 1]各第一 :意思是“这个维度中的所有元素”。

看看数组切片在 numpy 中是如何工作的 here .