python panda Dataframe 计算每2行之间的平方距离

q3qa4bjr  于 2022-10-30  发布在  Python
关注(0)|答案(1)|浏览(154)

我有包含数字行的 Dataframe

Writer  |   A   |   B   |   C 
   1    |  0.2  |  0.4  |  0.6
   2    |  0.1  |  1.2  |  0.3
   3    |  0.5  |  0.2  |  0.8

我想计算两个作家之间距离平方
有没有简单的方法可以做到这一点?

nwlqm0z1

nwlqm0z11#

如果需要所有对之间的距离,可以使用sklearn.metrics.pairwise_distances,如下所示:

from sklearn.metrics import pairwise_distances

D = pairwise_distances (df [['A','B','C']].values)

如果您只想知道两个写入器之间的距离,请说1和2:

from sklearn.metrics.pairwise import euclidean_distances

D_12 = euclidean_distances(df[['A','B','C']].values[0,:], df[['A','B','C']].values[1,:])

相关问题