两组经纬度矩阵如何求得最近的索引

zhanghm2018

求助一个问题，使用python语言。我有两组经纬度，分别为a文件的经纬度（维度为375, 335）:lat_a, lon_a,b文件的经纬度(维度为2030, 1354)：lat_b, lon_b.设计一个程序，使得对于a文件的每一个经纬度，都找出其在b文件经纬度的位置索引。

zhanghm2018

提供经纬度矩阵在附件给大家

读取数据的方法：
import numpy as np
data = np.load("data.npz")
lat_a = data["lat_a"]
lon_a = data["lon_a"]
lat_b = data["lat_b"]
lon_b = data["lon_b"]

zhanghm2018

zhanghm2018 发表于 2024-2-29 15:02
提供经纬度矩阵在附件给大家

读取数据的方法：

如果不想花费金币下载数据的加我的qq 704806663

zhanghm2018

最容易想到的方法是遍历a文件的经纬度
计算一次的消耗时间，从而得到计算一次全部的索引耗时1小时。（个人不能接受如此长时间的计算，想找办法加速计算）

zhanghm2018

更新解法：
使用k-d树的算法可以很快的求出多个维度的点之间的最短距离。
distance 为最小距离，index为最小距离点的索引