首页 > 解决方案 > 对任意维度的任意 numpy 数组进行切片

问题描述

假设我有一个任意维度的 numpy ndarray,我想提取该数组的前两个正面切片。最好的方法是什么?例如,如果输入数组 A 是 5 维的,我想提取 A[:,:,0:2,1,1]。同样,如果是 3 维的,我想提取 A[:,:,0:2],如果是 6 维,我想提取 A[:,:,0:2,1,1,1] 等。 .

标签: pythonarraysnumpy

解决方案


推荐阅读