沿动态指定的轴切片numpy数组

一尘不染

沿动态指定的轴切片numpy数组

python

我想沿着特定的轴动态切片一个numpy数组。鉴于这种：

axis = 2
start = 5
end = 10

我想要达到相同的结果：

# m is some matrix
m[:,:,5:10]

使用这样的东西：

slc = tuple(:,) * len(m.shape)
slc[axis] = slice(start,end)
m[slc]

但是:值不能放在元组中，所以我不知道如何构建切片。

阅读 159

2020-12-20

共1个答案

一尘不染

我认为一种方法是使用slice(None)：

>>> m = np.arange(2*3*5).reshape((2,3,5))
>>> axis, start, end = 2, 1, 3
>>> target = m[:, :, 1:3]
>>> target
array([[[ 1,  2],
        [ 6,  7],
        [11, 12]],

       [[16, 17],
        [21, 22],
        [26, 27]]])
>>> slc = [slice(None)] * len(m.shape)
>>> slc[axis] = slice(start, end)
>>> np.allclose(m[slc], target)
True

我有一种模糊的感觉，我以前曾经为此使用过一个函数，但是现在似乎找不到了。

2020-12-20