使用scipy的`cdist`函数可高效、向量化地计算任意坐标点集的两两欧氏距离矩阵,避免显式循环,兼顾简洁性与性能。
在科学计算和机器学习中,经常需要对一组二维(或高维)坐标点快速构建完整的成对距离矩阵(例如用于聚类、图构建或核方法)。虽然NumPy本身不直接提供距离矩阵函数,但借助scipy.spatial.distance.cdist——一个专为向量化计算设计的高性能工具,可在一行代码内完*部计算,且底层基于优化的C实现,远快于Python循环或np.linalg.norm嵌套广播。
以下是一个完整示例:
import numpy as np from scipy.spatial.distance import cdist # 原始坐标列表(支持任意维度) l_coords = [(1, 2), (1.1, 2.2), (1.05, 1.9)]# 转为 NumPy 数组(必需:cdist 要求输入为 (n_samples, n_features) 形状) coords = np.array(l_coords) # 计算对称距离矩阵(默认使用欧氏距离) dist_matrix = cdist(coords, coords) print(dist_matrix) # 输出: # [[0. 0.2236068 0.1118034 ] # [0.2236068 0. 0.30413813] # [0.1118034 0.30413813 0. ]]
✅ 关键说明:
⚠️ 注意事项:
综上,scipy.spatial.distance.cdist 是计算坐标距离矩阵的首选向量化方案:简洁、可靠、高效,是NumPy生态中不可或缺的“隐形加速器”。