获取到 cktTree 中最近邻居的距离

Question

coord_list 是一个 numpyarray，带有一些城市的 x 和 y 坐标。现在我想在两个单独的numpyarrays中找到每个城市的所有最近邻居的索引，以及到半径内所有最近邻居的距离。我得到了索引，但我找不到任何方法来获取距离。我尝试使用 distance.cdist(coord_list, coord_list, 'euclidean')，但我不知道如何使用 ind 来获得与 的正确距离距离。我刚开始用 Python 编程。

def construct_graph_connections(coord_list, radius):

mytree = spatial.cKDTree(coord_list)
ind = mytree.query_ball_point(coord_list, radius)
dist = distance.cdist(coord_list, coord_list, 'euclidean')
return ind, distance

Answer 1

您只需要通过在这些索引中评估您的coord_list 来提供您刚刚计算出的点数：

dist = distance.cdist(coord_list, coord_list[ind], 'euclidean')