python - 计算一组坐标之间的距离

Question

是否有更有效的方法来计算给定点集之间的欧氏距离？

这是我使用的代码:

def all_distances(position):
    distances = np.zeros((N_circles, N_circles))
    for i in range(N_circles):
        for j in range(i, N_circles):
            distances[i][j]=calculate_distance(position[i], position[j])
    return distances

def calculate_distance(p1, p2):
    return math.sqrt((p1[0]-p2[0])**2+(p1[1]-p2[1])**2)

position 是一个包含N_circles 个点坐标的数组。

Answer 1

你可以使用 pdist和 squareform来自科学

from scipy.spatial.distance import pdist, squareform

distances = pdist(position, metric="euclidean")
distance_matrix = squareform(distances)