从pandas DF中的数据点获取每组的质心

时间:2017-12-07 12:04:33

标签: python pandas dataframe

我有一个数据框

id    code    lat    long
1      100    22.6    42.3
1      200    23.6    45.3
1      400    21.6    46.3
2      300    22.6    42.3
2      500    22.6    42.3
2      800    22.6    42.3
3      100    22.6    42.3

我想找到id列上的中心点分组, 并返回一个数据帧:

id    centre_lat   centre_long
1      xx.xx        yy.yy
2      xx.xx        yy.yy
3      xx.xx        yy.yy

由于id 3只有1个代码,因此相同的lat long是该id的质心。

1 个答案:

答案 0 :(得分:2)

IIUC:

In [136]: df.groupby('id', as_index=False)['lat','long'].mean()
Out[136]:
   id   lat       long
0   1  22.6  44.633333
1   2  22.6  42.300000
2   3  22.6  42.300000