用另一个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)])