我有一个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
我忍不住想一定有更好的办法!有什么建议吗?你知道吗
S码
Tags:
也可以使用
np.take
。这样你就可以更自然地在一行中完成。你知道吗注意事项:
:
切片表示法,而是用range
替换它例如:
你可以这样做:
相关问题 更多 >
编程相关推荐