欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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)

pandas计算的欧式距离矩阵
如果以x,y为特征,计算不同城市之间的距离,不需要写循环,借助scipy 包,直接运算

代码如下:

from scipy.spatial import distance_matrix
pd.DataFrame(distance_matrix(df.values, df.values), index=df.index,columns=df.index)

结果如下:

pandas计算的欧式距离矩阵

相关标签: python pandas