我想系统地访问一个numpy数组的轴。例如,假设我有一个数组由INT选择numpy数组轴
a = np.random.random((10, 10, 10, 10, 10, 10, 10))
# choosing 7:9 from axis 2
b = a[:, :, 7:9, ...]
# choosing 7:9 from axis 3
c = a[:, :, :, 7:9, ...]
如果我有高维数组,则键入冒号会变得非常重复。现在,我想一些功能choose_from_axis
这样
# choosing 7:9 from axis 2
b = choose_from_axis(a, 2, 7, 9)
# choosing 7:9 from axis 3
c = choose_from_axis(a, 3, 7, 9)
所以,基本上,我想用一个号码访问轴。我知道如何做到这一点的唯一方法是来回使用rollaxis
,但我正在寻找一种更直接的方式来做到这一点。
谢谢。这很棒。我还想知道是否可以对元组做同样的事情,因为numpy数组接受元组作为索引集。 –