python - 表示 numpy 省略号

Question

例如，我知道如何用 numpy 对数组进行切片

v[1, :, :]

现在我想要一个函数将切片作为输入(1,1,None)并返回v[1,:,:]

问题是我不知道如何表示椭圆

Answer 1

您可以用 表示椭圆Ellipsis。

这里提到

Answer 2

slice(None)

所有这些都是等价的：

v[1, :, :]
v[1, Ellipsis, Ellipsis]
v[1, slice(None), slice(None)]
v[(1, Ellipsis, Ellipsis)]
v[(1, slice(None), slice(None))]

最后两个是我需要的。