在两个单独的数据框中找到与所有点最近的点

时间:2020-09-28 09:15:37

标签: python dataframe kdtree

我想找到从数据框B到位于A的所有数据点的最近点;我用过KDtree。查询,但是我不确定在哪里写A和B,我写的如下,是正确的还是应该在A和B之间切换? 例如,如果一个数据帧具有点A1,A2,A3,我想从数据帧B中找到最接近A1,A2,A3的点。

kd_wo_f = KDTree(B.values)
    #print(kd_wo_f.query(df_m_nostr.values, k=1)[-1])
    df_m['nearest_point_index']=kd_wo_f.query(A.values, k=1)[-1].tolist()

0 个答案:

没有答案