返回从集合X到集合Y的top-k最近距离

Question

我有两组点X和Y

函数pairwise_distances_argmin_min，在这里找到：

https://scikit-learn.org/stable/modules/generated/sklearn.metrics.pairwise_distances_argmin_min.html

给出X到Y的top-1距离。但是，对于我的应用程序，我对top 2或3感兴趣，并且计算从X到Y的每个元素之间的完整距离矩阵太昂贵了。

解决这个问题的最佳方法是什么？

预先感谢

Answer 1

工作评论：

scipy cKDTree, maybe? – Benjamin 20 hours ago

cKDTree的构建速度非常快，对于前2个邻居来说查询时间也不太长。具有32个维度的10000个数据点。