Python - 如何沿给定轴寻址数组

Question

我有一个从 3D 数组中读取切片的 python 脚本，如下所示：

def get_from_array(axis, start, end, array):
    if axis == 0:
        slice = array[start:end, :, :]
    elif axis == 1:
        slice = array[:, start:end, :]
    elif axis == 2:
        slice = array[:, :, start:end]
    return slice

我不禁想，一定有更好的方法！有什么建议吗？

S

Answer 1

你可以做这样的事情：

idx = [slice(None)] * array.ndim
idx[axis] = slice(start, end)
myslice = array[tuple(idx)]

Answer 2

您也可以

使用 np.take .然后你可以更自然地在一行中完成。

a.take(np.arange(start,end), axis=axis)

笔记：

• 它不接受标准的:切片表示法，而是将其替换为range
• 它返回副本，而不是视图

例如：

In [135]: a = np.arange(3*3*3).reshape(3,3,3)
In [136]: a.take(np.arange(1,2), axis=1)
Out[136]: 
array([[[ 3,  4,  5]],
       [[12, 13, 14]],
       [[21, 22, 23]]])