首页 > 解决方案 > 如何在只保留匹配的情况下加入数据框?

问题描述

我有两个数据框:

id   value
a1   12
a2   15
a3   17
id   value1
a1   22
a2   58
a7   10

我想通过列 id 加入它们,只保持匹配一个:

id   value    value1
a1    12       22
a2    15       58

怎么做?我试过 pd.merge 但它保留了所有 id

标签: pythonpython-3.xdataframejoinmerge

解决方案


您可以使用pd.mergeinner

数据框示例

df1 = pd.DataFrame({'id':['a1','a2', 'a3'],
                    'value':[12, 15, 17]})

df2 = pd.DataFrame({'id':['a1','a2', 'a7'],
                    'value':[22, 58, 10]})

代码

merged_df = df1.merge(df2, how = 'inner', on = ['id'])
merged_df.columns = ["id", "value", "value1"]

结果

   id  value  value1
0  a1     12      22
1  a2     15      58

如果要将 index 更改为id,可以使用pd.set_index('id')


推荐阅读