在计算数据框中的欧氏距离时如何保留索引信息?

时间:2015-07-26 23:53:03

标签: python pandas scipy

嗨我想用数据框中的X,Y坐标计算所有点之间的欧氏距离,并返回最近点的ID(索引)。

目前我正在使用它来创建距离矩阵:

diatancematrix=squareform(pdist(group))    
df=pd.DataFrame(dists)

然后返回最小点:

closest=df.idxmin()

我似乎无法在第一步中保留正确的ID /索引,因为它似乎从0开始分配列号和行号而不是使用索引。有没有办法在这里保持正确的索引?

1 个答案:

答案 0 :(得分:0)

距离矩阵包括每个点与其自身的距离,该距离始终为零。因此,您应该期望每一行都将自己视为自己的最小值。