python - pandas 自定义合并条件 key1 包含 key2

Question

我正在尝试合并两个数据帧，条件是 key1 包含 key2，示例如下

df1

name         info
Bill Gates   Microsoft
Steve Jobs   Apple
Elon Musk    Tesla

df2

name        info
Bill        aaa
Kate        bbb
Steve       ccc

我正在对键“name”列进行合并，我想要得到的结果是

df3

name          info         name     info
Bill Gates    Microsoft    Bill     aaa
Steve Jobs    Apple        Steve    ccc

在我的例子中，df1 相对较小，大约有 600 条记录，但 df2 相当大，大约有 200 万条记录，所以效率也是一个问题，非常感谢。

Answer 1

重命名内容并合并

rename_dict = {'name': 'full', 'info': 'company'}
df1.rename(columns=rename_dict).assign(
    name=lambda d: d.full.str.split().str[0]).merge(df2)

         full    company   name info
0  Bill Gates  Microsoft   Bill  aaa
1  Steve Jobs      Apple  Steve  ccc