任意尺寸的 Numpy 切片

Question

我想切片一個 numpy 數組以獲得最后一維中的第i個索引。 對於 3D 數組，這將是：

slice = myarray[:, :, i]

但是我正在編寫一個函數，我可以在其中獲取任意維度的數組，因此對於 4D 數組，我需要myarray[:, :, :, i]等等。 有沒有一種方法可以為任何數組獲取此切片，而無需顯式編寫數組維度？

Answer 1

有...或Ellipsis ，它正是這樣做的：

slice = myarray[..., i]

Ellipsis 是 python 對象，如果您想在方括號表示法之外使用它。

Answer 2

其實，剛剛找到答案。 正如numpy 的文檔中所述，這可以通過 slice 對象來完成。 在我的特殊情況下，這樣做可以：

idx = [slice(None)] * (myarray.ndim - 1) + [i] 
my_slice = myarray[idx]

slice(None)相當於選擇該索引中的所有元素，最后一個 [i] 為最后一個維度選擇特定索引。

Answer 3

在切片任意維度方面，之前的優秀答案可以擴展到：

indx = [slice(None)]*myarray.ndim
indx[slice_dim] = i
sliced = myarray[indx]

這將返回來自任何維度slice_dim的切片 - slice_dim = -1再現了先前的答案。 為了完整起見 - 上述清單的前兩行可以濃縮為：

indx = [slice(None)]*(slice_dim) + [i] + [slice(None)]*(myarray.ndim-slice_dim-1)

雖然我發現以前的版本更具可讀性。