python numpy库中省略号...的一些用法

在学习《Designing Machine Learning Systems with Python》(《机器学习系统设计Python语言实现》)一书的第五章梯度下降一节代码中（P105），代码中出现了x[..., 1] 这样子的代码，在之前的学习中我并没有遇到这个省略号，于是抱着试一试的态度，我运行了下代码，发现竟然可以运行，并输出了正确结果。这就很有意思了，我们常用的数据列表切片里的符号是冒号（形如x[:5] ）。省略号这还是第一次见。于是我仔细读了代码，发现代码中将x[..., 1] 换为x[:, 1] 并不会影响代码的结果。出于好奇我尝试了下该省略号是否可以用在所有列表数据类型呢？

>>>a = [i+1 for i in range(100)]
>>>a[..., 3]
Traceback (most recent call last):
  File "<input>", line 1, in <module>
TypeError: list indices must be integers or slices, not tuple

看来python自带的list类型并不能用省略号

>>>import numpy as np
>>>b = np.array(a)
>>>b[..., 1]
array(2)

哎，看来numpy库中的array类型数据是可以使用该省略号的。值得注意的是该b[..., 1] 返回的数据类型也是array类型。
于是我就去寻找了numpy库的说明文档，竟然让我找到了

Ellipsis expand to the number of : objects needed to make a selection tuple of the same length as x.ndim.
Only the first ellipsis is expanded, any others are interpreted as :.
Example

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

当然，手册中的这个例子x[...,0] 并不能替换为x[:,0] ，而是需要替换为x[:,:,0]


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

这个例子也可以让我们对省略号的用法更熟悉一些。