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


问题内容

我有类似的东西

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如果您能想到一个更精确的问题,请更改此问题的标题。


问题答案:

那用python呢?

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