用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的广播机制!

你可能感兴趣的:(Python)