如何找到連續的lat，lon點距離

Question

我有一個帶有一系列連續地理坐標的df。 我想找到這些連續點之間的距離。 1-> 2,2-> 3 .... end-1-> end。

使用df.shift(1)看起來並不漂亮，也使用循環。

使用一些遞歸函數可以更優雅地完成嗎？

Answer 1

解決方案

import pandas as pd

def calculate_distance(lat_from, long_from, lat_to, long_to):
    # some better logic
    return lat_from - lat_to + long_from - long_to

df = pd.DataFrame({'long': [1, 2, 4.2, 5, 6], 'lat': [7, 4, 2, 1.2, 2]})
df[['lat_to', 'long_to']] = df.shift(-1)

# this is way faster, but may not be possible depending on your calculation
calculate_distance(df['lat'], df['long'], df['lat_to'], df['long_to'])
>>> 0    2.000000e+00
>>> 1   -2.000000e-01
>>> 2    2.220446e-16
>>> 3   -1.800000e+00
>>> 4             NaN
>>> dtype: float64

# or
# a lot slower, processes on per-row basis
df.apply(lambda row: calculate_distance(row['lat'], row['long'], row['lat_to'], row['long_to']), axis=1)
>>> 0    2.000000e+00
>>> 1   -2.000000e-01
>>> 2    2.220446e-16
>>> 3   -1.800000e+00
>>> 4             NaN
>>> dtype: float64

要進行速度比較，請嘗試pandas.DataFrame.apply ， pandas.DataFrame.applymap和普通廣播操作之間的區別。