Python 3：2つのリストから最も近い数値の組み合わせを作りたい

Question

pythonを使って、2つのDataFrameのマッチングを行っています。
単に共通の値を基にマージするのではなく、値が近い組み合わせによってマージしたいです。例えば

df1
#       name   a
# 0        A  46
# 1        B  19
# 2        C   2

df2
#       name   b
# 0        D  50
# 1        E  40
# 2        F  10
# 3        G   4

に対して、(a,b)の組み合わせが

(2,4), (19,10), (46,50)

となるように導きたいです。
すなわち、Σ(a-b)^2が最小となるような組み合わせを効率よく導くにはどうしたらよいでしょうか。
ご教授ください。

Answer 1

pandas.merge_asofが使えます。

import pandas as pd

df1 = pd.DataFrame({'name':['A','B','C'],'a':[46,19,2]})
df2 = pd.DataFrame({'name':['D','E','F','G'],'b':[50,40,10,4]})

df1 = df1.sort_values('a')
df2 = df2.sort_values('b')

df = pd.merge_asof(df1,df2,left_on='a', right_on='b', direction='nearest')
print(df)
"""
  name_x   a name_y   b
0      C   2      G   4
1      B  19      F  10
2      A  46      D  50
"""

Comments

[takahashima]

ありがとうございます。pandasにそんな便利な関数があったんですね！もっと勉強しなくては…