Numpy中的快速图像坐标查找

时间:2012-10-02 03:34:51

标签: python numpy

我有一个充满坐标的大块状阵列(大约400个):

[[102, 234],
 [304, 104],
  ....
]

一个大小为800x800的ndy 2d数组my_map。

查找该数组中给出的坐标的最快方法是什么?我按照这篇文章中描述的那样尝试了调整等内容:http://opencvpython.blogspot.com/2012/06/fast-array-manipulation-in-numpy.html但无法让它发挥作用。

我还在考虑将每个坐标转换为地图的线性索引,然后将其直接导入my_map,如下所示:

my_map[linearized_coords]

但我无法通过矢量化将坐标正确转换为线性方式。有什么想法吗?

1 个答案:

答案 0 :(得分:2)

尝试使用tuple进行索引:

my_map[tuple(coords.T)]

这会通过advanced indexing选择数组的元素。