用numpy高效计算欧氏距离

本文介绍了一种利用numpy库高效计算多维空间中点间距离的方法。通过具体代码实例,展示了如何利用numpy的广播机制计算二维及更高维度的欧式距离,此方法直观且效率高。

在各类算法中,距离的计算极其常见,用numpy来计算效率非常的高,其计算方式也有很多。个人认为最直观也非常高效的一种方式如下:

首先看二维欧式距离的公式定义d=\sqrt{\left ( x_{1}-x_{2} \right )^{2}+\left ( y_{1}-y_{2} \right )^{2}},三维、四维等等继续扩展即可。

那么就依据公式的定义,我们直观的来设置计算公式,废话不多说直接见代码:

import numpy as np

a=np.arange(16).reshape(4,4)
x1=a[0]
x2=a[1].reshape(4,1)
y1=a[2]
y2=a[3].reshape(4,1)
d=np.sqrt((x1-x2)**2+(y1-y2)**2)

代码非常简单,而且高效,其实就是利用了numpy的广播机制!

### 计算欧氏距离的方法与公式 欧氏距离是一种常用的相似性度量方法,在多维空间中用于衡量两个点之间的直线距离。其基本定义如下: 对于 n 维空间中的两点 \( A(x_1, x_2, \ldots, x_n) \) 和 \( B(y_1, y_2, \ldots, y_n) \),它们的欧氏距离可以表示为: \[ d(A,B) = \sqrt{\sum_{i=1}^{n}(x_i - y_i)^2} \] #### 使用 Python 实现欧氏距离计算 以下是两种实现方式,分别通过手动编写函数以及调用 `numpy` 库来完成。 ##### 手动实现欧氏距离计算 可以通过简单的循环结构和平方根操作来实现欧氏距离计算[^3]: ```python import numpy as np def calEuclidean(x, y): dist = np.sqrt(np.sum(np.square(x - y))) return dist ``` 在此代码片段中,`np.array` 类型的数据可以直接进行向量、矩阵加减运算,而 `np.square` 则是对每个元素求平方后再累加并开方得到最终的距离值。 ##### 借助 NumPy 的内置功能 另一种更简洁的方式是利用 `numpy.linalg.norm` 函数直接计算范数作为欧氏距离: ```python import numpy as np def euclidean_distance_numpy(x, y): return np.linalg.norm(x - y) ``` 此方法内部实现了高效的数值计算逻辑,适用于大规模数据集场景下的性能优化需求。 #### SciPy 中的实现 除了上述基于 NumPy 的解决方案外,还可以借助于 SciPy 提供的功能快速获取两组或多组样本间的所有配对欧氏距离[^1]: ```python from scipy.spatial.distance import pdist, squareform Vec = [[1, 2], [4, 6]] dist_matrix = pdist(Vec, 'euclidean') result = squareform(dist_matrix) print(result) ``` 这里需要注意的是,当输入是一个二维数组时,`pdist` 返回的结果是一维压缩形式的距离矩阵;如果希望获得完整的方形距离表,则需进一步转换为标准格式。 --- ### 总结 综上所述,无论是采用基础数学表达式还是依赖第三方库的支持,都可以方便高效地完成欧氏距离的相关运算任务。具体选择取决于实际应用场景和个人偏好等因素影响下做出决定即可满足不同层次的要求[^2]。
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值