在Python中,我有一个包含数据的大2D数组,另一个mx22d数组包含感兴趣的m2d坐标集合,例如
coords=[[150, 123], [151, 123], [152, 124], [153, 125]]
我想提取包含这些坐标(索引)位置的数据数组值的Mx1数组。显然,data[coords]
不起作用。你知道吗
我怀疑有一个简单的方法可以做到这一点,但stackoverflow失败了我到现在为止。事先谢谢你的帮助。你知道吗
编辑:例如
data=[[0, 0, 0, 0, 0, 0, 1, 0],
[0, 0, 0, 1, 2, 1, 0, 0],
[0, 0, 0, 1, 23, 40, 0, 0],
[0, 0, 0, 1, 1, 2, 0, 0],
[0, 0, 3, 2, 0, 0, 0, 0],
[0, 0, 4, 5, 6, 2, 1, 0],
[0, 0, 0, 0, 1, 20, 0, 0],
[0, 0, 0, 3, 1, 2, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0]]
coords=[[1,4],[2,4],[2,5],[5,3],[6,5]]
期望的输出是
out=[2,23,40,5,20]
您可以使用list comprehension:
列表理解返回的结果等价于
相关问题 更多 >
编程相关推荐