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

Question

coord_list是一个在某些城市具有x和​​y坐标的numpyarray。现在，我想查找两个城市中每个城市的所有最近邻居的索引，还要找到两个单独的numpyarrays中半径内所有最近邻居的距离。我得到了索引，但是找不到找到距离的任何方法。我尝试使用 distance.cdist（coord_list，coord_list，'euclidean'），但我不知道如何使用 ind 从 dist 。我刚刚开始用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

您只需要通过在这些索引中评估distance.cdit()来给coord_list分配点：

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