python - 如何在 Python 中使用省略号切片语法？

Question

这出现在 Hidden features of Python ，但我看不到很好的文档或示例来解释该功能的工作原理。

Answer 1

省略号在 numpy 中用于对高维数据结构进行切片。

它的设计意思是此时，插入尽可能多的完整切片 (:) 以将多维切片扩展到所有维度。

示例:

>>> from numpy import arange
>>> a = arange(16).reshape(2,2,2,2)

现在，您有一个 2x2x2x2 阶的 4 维矩阵。要选择第 4 维中的所有第一个元素，可以使用省略号表示法

>>> a[..., 0].flatten()
array([ 0,  2,  4,  6,  8, 10, 12, 14])

相当于

>>> a[:,:,:,0].flatten()
array([ 0,  2,  4,  6,  8, 10, 12, 14])

在您自己的实现中，您可以随意忽略上述契约(Contract)并将其用于您认为合适的任何事情。