您所描述的也是 k-最近邻方法中的关键步骤。所以不需要重新发明轮子，我们可以看看其他人是如何加速该算法的。

我不知道任何像字典这样直接返回它的结构，但你可以使用 kd 树。如果实施得当，您可以很快获得 k 个最接近的向量。

这个 问题也可能很有趣。