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

发布于 2021-01-29 16:22:03

我有类似的东西

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

关注者
0
被浏览
48
1 个回答
  • 面试哥
    面试哥 2021-01-29
    为面试而生,有面试问题,就找面试哥。

    那用python呢?

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


知识点
面圈网VIP题库

面圈网VIP题库全新上线,海量真题题库资源。 90大类考试,超10万份考试真题开放下载啦

去下载看看