从向量中找到最接近的数字索引

Question

如果有参考向量，例如：ref = np.array([1., 3., 5.])和一个随机向量：r = np.array([3.1, 4.7, 0.1, 5.5])对于'r'中的每个数字，有没有一种快速的方法可以找到'ref'中关闭数字的索引？预期结果是：[2, 3, 1, 3]3.1 最接近 3，所以答案是索引 2。4.7 最接近 5，所以答案是指数 3。0.1 最接近 1，所以答案是索引 1。5.5 最接近 5，所以答案是指数 3。

Answer 1

你可以做一个广播：

np.abs(r[:,None]-ref).argmin(-1)

输出（记住 python 是从 0 开始索引的）：

array([1, 2, 0, 2])