我得到了一个包含两列的数据框,其中包含经度和纬度坐标:
将 pandas 导入为 pd
values = {'Latitude': {0: 47.021503365600005,
1: 47.021503365600005,
2: 47.021503365600005,
3: 47.021503365600005,
4: 47.021503365600005,
5: 47.021503365600005},
'Longitude': {0: 15.481974060399999,
1: 15.481974060399999,
2: 15.481974060399999,
3: 15.481974060399999,
4: 15.481974060399999,
5: 15.481974060399999}}
df = pd.DataFrame(values)
df.head()
现在我想在数据帧上应用滚动窗口函数,该函数采用一行和另一行(窗口大小 2)的经度和纬度(两列)来计算半正矢距离。
def haversine_distance(x):
print (x)
df.rolling(2, axis=1).apply(haversine_distance)
我的问题是我从未获得所有四个值 Lng1、Lat1(第一行)和 Lng2、Lat2(第二行)。如果我使用 axis=1,那么我将得到第一行的 Lng1 和 Lat1。如果我使用 axis=0,那么我将获得第一行和第二行的 Lng1 和 Lng2,但仅获得经度。
如何使用两行和两列应用滚动窗口?有点像这样:
def haversine_distance(x):
row1 = x[0]
row2 = x[1]
lng1, lat1 = row1['Longitude'], row1['Latitude']
lng2, lat2 = row2['Longitude'], row2['Latitude']
# do your stuff here
return 1
目前,我正在通过 shift(-1) 将数据帧与其自身连接起来进行此计算,从而将所有四个坐标放在一行中。但滚动也应该是可能的。另一种选择是将 Lng 和 Lat 合并到一列中,并在其上应用 axis=0 的滚动。但一定有更简单的方法,对吗?