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

2024-04-19 05:20:06 发布

您现在位置:Python中文网/ 问答频道 /正文

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

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

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

然后返回最小点:

closest=df.idxmin()

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


Tags: 数据id距离dataframedfgroup矩阵pd