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

最新推荐文章于 2025-06-26 15:57:05 发布

清风不识字12138

最新推荐文章于 2025-06-26 15:57:05 发布

阅读量1.3w

点赞数 14

CC 4.0 BY-SA版权

分类专栏： python学习机器学习系统实现文章标签： python numpy

本文链接：https://blog.youkuaiyun.com/qq_33363973/article/details/79894982

python学习同时被 2 个专栏收录

21 篇文章

订阅专栏

机器学习系统实现

4 篇文章

订阅专栏

本文探讨了Python编程语言中省略号(...)在NumPy数组索引中的作用及其与传统列表切片的区别。通过具体示例说明了省略号如何在多维数组中工作，并解释了其在不同上下文中的行为。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在学习《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]])

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