假设我们有一个多维数组。
import numpy as np
foo = np.random.random((2,4,3,5))
每个轴都与数据的特定功能相关,我对数据的一个子集感兴趣。
我可以在轴2上使用逻辑索引,在轴0和3上使用元素索引。
mask = np.array([1, 1, 0], dtype=bool)
bar = foo[0,:,mask,0]
我期望酒吧有形状(4,2)。相反,它有形状(2,4)。每个轴内的数据符合预期。然而,轴是从我期望的转换。
我怀疑这与播放面具有关,但我无法找到解决方法。
Python 3.6.4
Numpy 1.14.0