多维布尔索引到多维数组

Question

我正在尝试在多维数组中使用多维布尔索引。这是我正在尝试做的一个简单版本。

import numpy as np
pf=np.array([[[-67.  ,-20.35, 1],
              [-46.35, 33.25, 2],
              [  1   ,  2   , 1],
              [  4   ,  4   , 4]],
             [[-64.  ,-70.35, 8],
              [-46.35, 33.25, 7],
              [  3   ,  7   , 8],
              [  9   ,  3   , 2]]])
booly=np.array([[False, True, False, False],
                [False, False, False, True]])

我想使用我的布尔数组从我的初始 (2,4,3) 数组中选择真实值，以获得最终的 (2,3) 数组，而无需循环：

truth=np.array([[-46.35, 33.25, 2],
                [  9   ,  3   , 2]])

我尝试过但失败了。谢谢大家。

Answer 1

numpy 支持boolean indexing：

res = pf[booly]
print(res)
# [[-46.35  33.25   2.  ]
# [  9.     3.     2.  ]]