在学习《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]])
这个例子也可以让我们对省略号的用法更熟悉一些。