python - NumPy 中的数组索引没有意义

Question

我有一个大小为 28x28x60000 的 numpy 数组。请注意以下几点:

>>> X.shape
(28, 28, 60000)
>>> X[:][:][0].shape
(28, 60000)

X[:][:][0] 不应该是一个大小为 28x28 的数组吗？我们包括前两个维度中的每个组件(每个 28 个)，但仅包括第三个维度中的第 0 个条目。

这是怎么回事？

Answer 1

你切错了。 Slice X[:] 返回原始数组的副本。所以你的切片可以解释为 X 的第一个副本，然后是 X 的另一个副本，然后得到第一个具有 28x60000 形状的元素。所以你需要调用 X[:,:,0]。示例:

import numpy as np
X = np.random.randn(28,28,60000)

In [257]: X[:,:,0].shape
Out[257]: (28, 28)

您可以将您的 X[:] 和 X[:][:] 语句与 all 进行比较

In [261]: (X[:] == X[:][:]).all()
Out[261]: True