Как получить n-й ближайший сосед для каждой точки в массиве NumPy из того же массива?

Question

У меня есть массив с формой (291336, 50).то есть есть 291336 точек, в которых каждая точка имеет 50 измерений.

Для каждой точки в этом массиве я хочу найти расстояние и индекс k ее ближайшего соседа по расстоянию, принадлежащему тому же массиву.Я нашел этот связанный вопрос, но он находит 1st ближайшего соседа, а не k th.

Я думал об использовании этого подхода грубой силы -

for i in X.shape[0]:
    distance_from_i = {}
    for j in X.shape[0]:
        store distance & index of j from i in distance_from_i
    sort distance_from_i and select the k'th point

Но я знаю, что это ужасно.Должен быть лучший способ.

Как мне решить эту проблему?

Answer 1

Как насчет сортировки их по расстоянию от np.zeros (50)