我有一个 python 脚本,它从 3D 数组中读取切片,如下所示:
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
我不禁认为必须有更好的方法来做到这一点!有什么建议么?
小号