用另一个2D数组索引NumPy 2D数组

Question

我有类似的东西

m = array([[1, 2],
            [4, 5],
            [7, 8],
            [6, 2]])

和

select = array([0,1,0,0])

我的目标是

result = array([1, 5, 7, 6])

我在simplfy行和列提取numpy中阅读时尝试了_ix，但这并没有达到我想要的结果。

ps如果您能想到一个更精确的问题，请更改此问题的标题。

Answer 1

那用python呢？

result = array([subarray[index] for subarray, index in zip(m, select)])