pandas计算的欧式距离矩阵
程序员文章站
2022-07-12 19:55:31
...
通常我们用pandas 保存数据 ,对于某些属性,需要计算不同对象之间的欧式距离;
数据产生,如下所示:
import pandas as pd
ctys = ['Boston', 'Phoenix', 'New York']
df = pd.DataFrame(data, columns=['xcord', 'ycord'], index=ctys)
如果以x,y为特征,计算不同城市之间的距离,不需要写循环,借助scipy 包,直接运算
代码如下:
from scipy.spatial import distance_matrix
pd.DataFrame(distance_matrix(df.values, df.values), index=df.index,columns=df.index)
结果如下:
上一篇: 1、webpack4.0 基础配置
下一篇: webpack插件和配置