为什么元素和逻辑索引数组的形状似乎转换尺寸?

时间:2018-05-09 15:50:21

标签: python numpy numpy-broadcasting

假设我们有一个多维数组。

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

0 个答案:

没有答案