在各类算法中,距离的计算极其常见,用numpy来计算效率非常的高,其计算方式也有很多。个人认为最直观也非常高效的一种方式如下:
首先看二维欧式距离的公式定义d=,三维、四维等等继续扩展即可。
那么就依据公式的定义,我们直观的来设置计算公式,废话不多说直接见代码:
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的广播机制!