python - 为什么用括号和逗号索引 numpy 数组的行为不同？

Question

我倾向于用方括号索引 numpy 数组(矩阵)，但我注意到当我想对数组(矩阵)进行切片时，我必须使用逗号表示法。为什么是这样？例如，

>>> x = numpy.array([[1, 2], [3, 4], [5, 6]])
>>> x
array([[1, 2],
       [3, 4],
       [5, 6]])
>>> x[1][1]
4                 # expected behavior
>>> x[1,1]
4                 # expected behavior
>>> x[:][1]
array([3, 4])     # huh?
>>> x[:,1]
array([2, 4, 6])  # expected behavior

Answer 1

这个:

x[:, 1]

表示“沿第一个轴取 x 的所有索引，但沿第二个轴只取索引 1”。

这个:

x[:][1]

表示“沿第一个轴取 x 的所有索引(所以所有 x)，然后沿 first 轴取索引 1结果的”。您将 1 应用于错误的轴。

x[1][2] 和 x[1, 2] 只是等价的，因为用整数索引数组会将所有剩余的轴移向前面形状，所以 x[1] 的第一个轴是 x 的第二个轴。这根本不能一概而论；您应该几乎总是使用逗号而不是多个索引步骤。