python - numpy 数组切片意外结果

Question

我不明白下面的行为。 numpy 数组通常可以通过索引访问，所以 [:,1] 应该等同于 [:][1]，我是这么认为的。有人可以解释为什么不是这样吗？

>>> a = np.array([[1, 2, 3], [4, 5, 6]])  
>>> a[:,1]  
array([2, 5])  
>>> a[:][1]  
array([4, 5, 6])

谢谢!

Answer 1

这两种形式的索引是不一样的。您应该使用 [i, j] 而不是 [i][j]。即使两者都有效，第一个也会更快(参见 this question )。

使用两个索引 [i][j] 是两个操作。它执行第一个索引，然后根据第一个操作的结果执行第二个索引。 [:] 仅返回整个数组，因此您的第一个等同于 array[1]。由于只传递了一个索引，它假定引用第一个维度(行)，所以这意味着“获取第 1 行”。使用一个复合索引 [i, j] 是同时使用两个索引条件的单个操作，因此 array[:, 1] 返回“所有行，第 1 列” .