python - 任意尺寸的 Numpy 切片

Question

我想对一个 numpy 数组进行切片以获得最后一维中的第 i-th 索引。对于 3D 数组，这将是:

slice = myarray[:, :, i]

但我正在编写一个函数，我可以在其中获取任意维度的数组，因此对于 4D 数组，我需要 myarray[:, :, :, i] 等等。有没有一种方法可以在无需显式写入数组维度的情况下为任何数组获取此切片？

Answer 1

有 ... 或 Ellipsis，正是这样做的:

slice = myarray[..., i]

Ellipsis 是 Python 对象，如果您想在方括号表示法之外使用它。