熊猫在地理距离中使用前一行的值答案

【问题标题】：Pandas use the value from previous row in a geopy distance熊猫在地理距离中使用前一行的值
【发布时间】：2018-12-21 09:12:06
【问题描述】：

我正在尝试根据上一行的坐标计算地理距离。有没有一种方法可以在不向数据框添加额外列的情况下进行计算？

示例代码：

import pandas
import geopy.distance

d = {'id_col':['A','B','C','D'], 
  'lat':[ 40.8397,40.7664,40.6845,40.6078], 
  'lon':[-104.9661,-104.999,-105.01,-105.003]
   }
df = pandas.DataFrame(data=d)

lambda 和 apply 的第一种方法

df['geo_dist']=df.apply(lambda x: geopy.distance.geodesic((x['lat'],x['lon']),(x['lat'].shift(),x['lon']).shift()),axis=1)

我会收到错误：AttributeError: ("'float' object has no attribute 'shift'", u'occurred at index 0')

我的第二种方法是在数据帧上调用函数：

def geodist(x):
    return geopy.distance.geodesic((x['lat'],x['lon']),(x['lat'].shift(),x['lon']).shift())

df['geo_dist']=geodist(f)

在这种情况下，我会收到错误消息：ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().

非常感谢任何帮助。

【问题讨论】：

我无法导入geopy.distance，但我认为您需要使用df.shift().lat

标签： python pandas geopy

【解决方案1】：

第一种方法不会那样工作，因为 lambda 函数应用于 DataFrame 的单行，并且 x 不是您期望的所有观察结果的列表。为此，您可以使用 x.name-1 获取前一个元素索引并访问 df 中的位置，如下所示

df['geo_dist']=df.apply(lambda x: geopy.distance.geodesic((x['lat'],x['lon']),(df.iloc[x.name - 1].lat,df.iloc[x.name - 1].lon)) if x.name > 0 else 0,axis=1)

希望对你有帮助

【讨论】：