使用GeoPy计算熊猫数据框上的纬度/经度距离

时间:2020-08-04 20:26:55

标签: python pandas geopy

我正在尝试使用Pandas数据框上的geopy计算纬度和经度之间的距离。

这是我的数据框:

    latitude    longitude   altitude
    -15.836310  -48.020298  1137.199951
    -15.836360  -48.020512  1136.400024
    -15.836415  -48.020582  1136.400024
    -15.836439  -48.020610  1136.400024
    -15.836488  -48.020628  1136.599976

我尝试了两种不同的方式:

from geopy import distance

for i in range(1, len(df)):
   before = (df.loc[i-1, 'latitude'], df.loc[i-1, 'longitude'])
   actual = (df.loc[i, 'latitude'], df.loc[i, 'longitude'])
   df.loc[i, 'geodesic'] = distance.distance(before, actual).miles

错误:

 KeyError: 0

显然,df.loc[i, 'column_name']不起作用。

和:

from geopy import distance

df['geodesic'] = distance.distance((df.latitude.shift(1), df.longitude.shift(1)), (df.latitude, df.longitude)).miles

错误:

ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().

官方GeoPy文档:

from geopy import distance
newport_ri = (41.49008, -71.312796)
cleveland_oh = (41.499498, -81.695391)
print(distance.distance(newport_ri, cleveland_oh).miles)

2 个答案:

答案 0 :(得分:1)

我得到了错误。

1-我必须检查latitudelongitudeNaN

2-我无法将time设置为索引。 (我不知道为什么,这花了很长时间才发现)

一旦检查,错误就消失了。

答案 1 :(得分:1)

raw = """latitude;longitude;altitude
-15.836310;-48.020298;1137.199951
-15.836360;-48.020512;1136.400024
-15.836415;-48.020582;1136.400024
-15.836439;-48.020610;1136.400024
-15.836488;-48.020628;1136.599976"""

import pandas as pd
from io import StringIO
from geopy import distance

data = StringIO(raw)
df = pd.read_csv(data, sep=";")
df1 = df.drop(['altitude'], axis=1)
locations = df1.apply(tuple, axis=1)

for counter in range(len(locations) - 1):
    print(distance.distance(locations[counter], locations[counter + 1]).miles)

来自df = pd.read_csv(data, sep=";"),它与您的代码相同,我使其可测试。

然后,df1 = df.drop(['altitude'], axis=1)放下工作台的z轴,此应用程序不需要。

将df1转换为元组,并在各个位置之间循环,即可获得距离