多維布爾索引到多維數組

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支持布爾索引 ：

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