这个 Slicing 和 Ellipse 索引操作会导致什么结果？

Question

我尝试使用 ndarrays 了解这段代码会发生什么：

max_evecs = evecs[..., :, 0]

其中 evecs 是 ndarray 类型。到目前为止，我知道 ':' 是一个切片运算符，而 '...' 是一个所谓的椭圆。到目前为止，':' 表示所有元素，而 '...' 表示需要的元素。我调查了形状的数组：

>>> max_evecs.shape
(128, 128, 72)
>>> evecs.shape
(128, 128, 72, 3)

我可以想象，它试图将 3D 数组转换为三元组 (x,y,z) 数组。但我不确定。

对于那些关心的人：它来自 dipy 软件包：函数 quantize_evecs 内 https://github.com/nipy/dipy/blob/ff75b192f694cdb62cc11310159cdb652ce62073/dipy/reconst/dti.py 1663 附近的行

Answer 1

来自http://docs.scipy.org/doc/numpy/reference/arrays.indexing.html

省略号扩展为 : 进行选择所需的对象数 与 x.ndim 长度相同的元组。只有第一个省略号是 展开后，任何其他都被解释为：。

所以在您的示例中，evecs 的 ndim 等于 4，以下是等价的：

evecs[..., :, 0]
evecs[:, :, :, 0]

以交互方式进行尝试以感受它也很有帮助。使用一些简单的数据，例如 a = np.random.rand(3, 3, 2)，尝试打印出 a 和它的各个部分，如 a[..., 0] 和 a[1, ...] 和类似的，看看它们是如何相关的。