取得最近点的有效方法

Question

我有两个点列表，它们存储为2d数组的行：

a = np.array([[1, 2], [3, 4], [5, 6]])
b = np.array([[1, 2], [3, 4]])

对于b中的每个点，我想获取a中最近的点的索引。目前，我的解决方案是

f = lambda x: np.argmin(np.linalg.norm(a - x, axis=1))
print(list(map(f, b)))

有没有更有效的方法？我已经尝试过np.vectorize(f)(b)，但没有用。