Python pandas -基于纬度和经度的距离聚合列

import pandas as pd
import math

# 创建一个示例数据集
data = {
    '经度': [116.397128, 117.197978, 113.943649, 121.618622, 113.3246],
    '纬度': [39.916527, 39.137393, 22.549295, 38.914611, 23.1317]
}

df = pd.DataFrame(data)

# 计算两点之间的距离
def calculate_distance(lat1, lon1, lat2, lon2):
    radius = 6371  # 地球半径，单位为公里
    dlat = math.radians(lat2 - lat1)
    dlon = math.radians(lon2 - lon1)
    a = math.sin(dlat / 2) * math.sin(dlat / 2) + math.cos(math.radians(lat1)) * math.cos(math.radians(lat2)) * math.sin(
        dlon / 2) * math.sin(dlon / 2)
    c = 2 * math.atan2(math.sqrt(a), math.sqrt(1 - a))
    distance = radius * c
    return distance

df['距离'] = df.apply(lambda x: calculate_distance(x['纬度'], x['经度'], 39.9, 116.4), axis=1)

# 打印结果
print(df)