вернуть top-k самое близкое расстояние от множества X до множества Y

Question

У меня есть 2 набора точек, X и Y

Функция pairwise_distances_argmin_min, найденная здесь:

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

дает расстояние топ-1 от X до Y. Однако для моего приложения меня интересуют топ-2 или 3 и вычисление матрицы полного расстояния между каждым элементом изX к Y слишком дорогой.

Какой лучший способ это сделать?

заранее спасибо

Answer 1

рабочий комментарий:

scipy cKDTree, maybe? – Benjamin 20 hours ago

CKDTree создается очень быстро, и время запроса не слишком велико для соседей top2.С 10000 точками данных 32 измерений.