pandas内部连接数据帧而不合并数据

时间:2015-12-01 22:19:11

标签: python python-2.7 pandas inner-join

我有2个索引数据帧(第0列索引):

0       1       2          3
JoeK    Joe     Kavanagh   joe.kavanagh@nomail.com
BarryD  Barry   Dempsy     bdempsy@nomail.com
OrlaF   Orla    Farrel     ofjk@nomail.com
SethB   Seth    Black      sblack@nomail.com
KateW   Kate    White      kw12@nomail.com

和第二个:

0       1       2          3
JoeK    Joe     Kavanagh   jkavanagh@nomail.com
BarryD  Barry   Dempsy     barry.dempsy@nomail.com
JimmyS  Jimmy   Smith      j.Smith@nomail.com
AndyB   Andy    Brown      ABrwn@nomail.com
MaryP   Mary    Power      MaryPower@nomail.com

我想执行内部联接,如下面的SQL:

SELECT df2.* FROM df2
INNER JOIN df1
ON df2.0 = df1.0

我只从第二个数据框得到结果,而不是两者:

0       1       2          3
JoeK    Joe     Kavanagh   jkavanagh@nomail.com
BarryD  Barry   Dempsy     barry.dempsy@nomail.com

我尝试过pandas merge,但是它给了我两个数据帧的结果! 非常感谢任何帮助。

2 个答案:

答案 0 :(得分:0)

而不是合并,你想通过使用isin测试值与其他df的成员身份来过滤你的第二个df:

In [16]:
df1[df1['0'].isin(df['0'])]

Out[16]:
        0      1         2                        3
0    JoeK    Joe  Kavanagh     jkavanagh@nomail.com
1  BarryD  Barry    Dempsy  barry.dempsy@nomail.com

答案 1 :(得分:0)

你需要指定你想要一个内部 merge(默认是做一个外部)。您还需要先将df1限制为合并列:

In [11]: df2.merge(df1[['0']], how="inner", on=['0'])  # equivalently df1[['0']].merge(df2, how="inner", on=['0'])
Out[11]:
        0      1         2                        3
0    JoeK    Joe  Kavanagh     jkavanagh@nomail.com
1  BarryD  Barry    Dempsy  barry.dempsy@nomail.com

如果您不执行限制(df1[['0']]),它将为重叠列添加后缀:

In [12]: df2.merge(df1, how="inner", on=['0'])
Out[12]:
        0    1_x       2_x                      3_x    1_y       2_y                      3_y
0    JoeK    Joe  Kavanagh     jkavanagh@nomail.com    Joe  Kavanagh  joe.kavanagh@nomail.com
1  BarryD  Barry    Dempsy  barry.dempsy@nomail.com  Barry    Dempsy       bdempsy@nomail.com

后缀可以使用suffixes kwarg配置。

See also the pandas docs for a "brief primer on merge methods"