任意尺寸的块状切片

Question

我想切片一个numpy数组以获得最后一个维度中的第i个索引。对于3D阵列，这将是：

slice = myarray[:,:,i]

但是我正在编写一个函数，可以获取任意维度的数组，因此对于4D数组，我需要myarray[:,:,:,i]，依此类推。有没有一种方法可以获取任何数组的切片，而无需显式编写数组维？

Answer 1

存在...或Ellipsis，它正是这样做的：

slice = myarray[..., i]

如果您想在方括号符号之外使用Ellipsis，则它是python对象。