python - 如何使用多列合并两个数据框？

Question

我有两个数据框:

df_1 = pd.DataFrame(columns=["pointid","lat","lon"],data=[[1,41.792145,3.046884],[2,41.799834,3.051082],[3,41.813694,3.063463], [4,41.817673,3.067025]])

df_2 = pd.DataFrame(columns=["id","point_from","point_to"],
                              data=[[1,1,2],[2,1,3],[3,2,3]])

我想合并它们以获得一个新的数据框 df_3 与以下列(1 行的示例):

id    point_from    point_to   lat_from   lon_from   lat_to     lon_to
1     1             2          41.792145  3.046884   41.799834  3.051082

我该怎么做？

Answer 1

一种选择是分别在 point_from 和 point_to 列上将 df_1 与 df_2 合并两次:

df_1 = df_1.set_index("pointid")

(df_2.merge(df_1.add_suffix("_from"), left_on="point_from", right_index=True)
     .merge(df_1.add_suffix("_to"), left_on="point_to", right_index=True))

#   id  point_from  point_to   lat_from  lon_from     lat_to    lon_to
#0   1           1         2  41.792145  3.046884  41.799834  3.051082
#1   2           1         3  41.792145  3.046884  41.813694  3.063463
#2   3           2         3  41.799834  3.051082  41.813694  3.063463