从numpy中的任意维度删除行

Question

我有一个函数remrow，它将任意numpy nd 数组arr 和一个整数n 作为输入。我的函数应该从arr 中删除nth 维度中的最后一行。例如，如果像这样调用我的函数：

remrow(arr,2)

arr 作为 3d 数组，那么我的函数应该返回：

arr[:,:,:-1]

如果我打电话也是如此；

remrow(arr,1)

而arr 是一个5d 数组，那么我的函数应该返回：

arr[:,:-1,:,:,:]

我的问题是这样的；我的功能必须适用于arr 的所有形状和大小以及所有兼容的n。如何使用 numpy 数组索引来做到这一点？

Answer 1

构造一个索引元组，由 slice(None) 和 slice(None,-1) 对象的所需组合组成。

In [75]: arr = np.arange(24).reshape(2,3,4)
In [76]: idx = [slice(None) for _ in arr.shape]
In [77]: idx
Out[77]: [slice(None, None, None), slice(None, None, None), slice(None, None, None)]
In [78]: idx[1]=slice(None,-1)
In [79]: arr[tuple(idx)].shape
Out[79]: (2, 2, 4)
In [80]: idx = [slice(None) for _ in arr.shape]
In [81]: idx[2]=slice(None,-1)
In [82]: arr[tuple(idx)].shape
Out[82]: (2, 3, 3)