python - 合并与函数输出相关的 Pandas 数据帧

Question

是否有一种方便的方法可以根据行之间的距离合并两个数据帧？对于以下示例，我想从最近的 df2 行中获取 df1 行的颜色。距离应计算为 ((x1-x2)**0.5+(y1-y2)**0.5)**0.5 .

import pandas as pd

df1 = pd.DataFrame({'x': [50,16,72,61,95,47],'y': [14,22,11,45,58,56],'size':[1,4,3,7,6,5]})
df2 = pd.DataFrame({'x': [10,21,64,31,25,55],'y': [54,76,68,24,34,19],'color':['red','green','blue','white','brown','black']})

Answer 1

来自 numpy 的东西播送

df1['color']=df2.color.iloc[np.argmin(np.sum(np.abs(df1[['x','y']].values-df2[['x','y']].values[:,None])**0.5,2),0)].values
df1
Out[79]: 
    x   y  size  color
0  50  14     1  black
1  16  22     4  white
2  72  11     3  black
3  61  45     7   blue
4  95  58     6   blue
5  47  56     5    red