测量点和组之间的距离

2024-06-16 08:50:24 发布

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

我试图测量熊猫数据帧内点之间的距离。我首先要测量一个子区域中的点之间的距离,然后得到该组的平均距离。然后我要测量子区域之间的距离(测量这两个向量之间的距离)。我知道如何做测量部分(前者使用scipy.spatial.distance.euclidean,后者使用scipy.spatial.distance.cdist)。我遇到的问题是如何将函数应用于数据集。我想我应该用groupby.apply()并输入我的函数,但我很难将其概念化。数据帧如下所示:

id, latitude, longitude, subregion, region

目前我有:

^{pr2}$

我知道这是不正确的,因为我想对所有行应用多个列。我的另一个想法是我使用了错误的数据结构。在


Tags: 数据函数id区域距离scipy向量spatial
1条回答
网友
1楼 · 发布于 2024-06-16 08:50:24

最后,我使用了不同的数据结构,结果如下:

contacts = {}

for i, row in sc_walkbook.iterrows():
    if contacts.get(row['region'],0) == 0:
        contacts[row['region']] = {}
        contacts[row['region']][row['subregion']] = {}
        contacts[row['region']][row['subregion']]['coords'] = []
        contacts[row['region']][row['subregion']]['distances'] = []
    elif contacts[row['region']].get(row['subregion'],0) == 0:
        contacts[row['region']][row['subregion']] = {}
        contacts[row['region']][row['subregion']]['coords'] = []
        contacts[row['region']][row['subregion']]['distances'] = []
    else:
        pass
    contacts[row['region']][row['subregion']]['coords'].append([row['T_Latitude'],row['T_Longitude']])

for region in contacts.itervalues():
    for subregion in region.itervalues():
        for a, b in itertools.combinations(subregion['coords'], 2):
            subregion['distances'].append(euclidean(a, b))

相关问题 更多 >