Ellipsis在NumPy索引中的含义

Ellipsis在NumPy索引中的含义

Content #

Ellipsis会扩展成多个 :,其个数会保证原数组所有的维度上都会被选择。

>>> x = np.array([[[1],[2],[3]], [[4],[5],[6]]])
>>> x.shape
(2, 3, 1)
>>> x[..., 0]
array([[1, 2, 3],
      [4, 5, 6]])

x有三个维度,因此Ellipsis会被扩展成两个 :,以保证每个维度都被选择。

From #