计算数据帧中纬度和经度之间的距离

时间:2017-06-08 23:03:54

标签: python pandas geopy

我的数据框中有4列包含以下数据:
Start_latitude
Start_longitude
Stop_latitude
Stop_longitude

我需要计算纬度经度对之间的距离,并创建一个计算距离的新列。

我遇到了一个可以为我做这件事的包裹(geopy)。但我需要将元组传递给geopy。如何在pandas中的数据框中为所有记录应用此函数(geopy)?

2 个答案:

答案 0 :(得分:8)

我建议您使用pyproj而不是geopy。 geopy依赖于在线服务,而pyproj是本地的(意味着它会更快,不会依赖于互联网连接),并且对其方法更加透明(例如参见here),它们基于Proj4代码库基本上是所有开源GIS软件的基础,也可能是您使用的许多Web服务。

#!/usr/bin/env python3

import pandas as pd
import numpy as np
from pyproj import Geod

wgs84_geod = Geod(ellps='WGS84') #Distance will be measured on this ellipsoid - more accurate than a spherical method

#Get distance between pairs of lat-lon points
def Distance(lat1,lon1,lat2,lon2):
  az12,az21,dist = wgs84_geod.inv(lon1,lat1,lon2,lat2) #Yes, this order is correct
  return dist

#Create test data
lat1 = np.random.uniform(-90,90,100)
lon1 = np.random.uniform(-180,180,100)
lat2 = np.random.uniform(-90,90,100)
lon2 = np.random.uniform(-180,180,100)

#Package as a dataframe
df = pd.DataFrame({'lat1':lat1,'lon1':lon1,'lat2':lat2,'lon2':lon2})

#Add/update a column to the data frame with the distances (in metres)
df['dist'] = Distance(df['lat1'].tolist(),df['lon1'].tolist(),df['lat2'].tolist(),df['lon2'].tolist())

PyProj有一些文档here

答案 1 :(得分:3)

来自geopy的文档:https://pypi.python.org/pypi/geopy。你可以这样做:

from geopy.distance import vincenty

# Define the two points
start = (start_latitute, start_longitude)
stop = (stop_latitude, stop_longitude)

# Print the vincenty distance
print(vincenty(start, stop).meters)

# Print the great circle distance
print(great_circle(start, stop).meters)

将此与熊猫相结合。假设您有一个数据框df。我们首先创建函数:

def distance_calc (row):
    start = (row['start_latitute'], row['start_longitude'])
    stop = (row['stop_latitude'], row['stop_longitude'])

    return vincenty(start, stop).meters

然后将其应用于数据帧:

df['distance'] = df.apply (lambda row: distance_calc (row),axis=1)

注意axis = 1说明符,这意味着应用程序是在一行而不是列级别完成的。