python - Numpy 中的快速图像坐标查找

Question

我有一个充满坐标的大 numpy 数组（大约 400 个）：

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

还有一个大小为 800x800 的 numpy 二维数组 my_map。

查找该数组中给出的坐标的最快方法是什么？我尝试了像这篇文章中描述的调色板：http://opencvpython.blogspot.com/2012/06/fast-array-manipulation-in-numpy.html但无法让它工作。

我还在考虑将每个坐标转换为地图的线性索引，然后将其直接通过管道传输到 my_map 中，如下所示：

my_map[linearized_coords]

但我无法矢量化以将坐标正确转换为线性方式。有任何想法吗？

Answer 1

尝试使用tuple索引：

my_map[tuple(coords.T)]

这通过高级索引选择数组元素。