使用Pandas“isin”-syntax进行子集选择

时间:2015-10-28 13:29:27

标签: python pandas subset

我有一个关于表的问题(表A - 包含三个键的多个值和一些“值”列),如下所示:

ID     TIME1       TIME2     VALUE_A      VALUE_B
1      201501      201501    a            1a
1      201502      201502    a            1c
1      201502      201502    b            1d 
1      201501      201501    b            2e
1      201501      201501    b            6a
1      201501      201501    b            1d 
1      201502      201502    b            2e
1      201502      201502    b            6a

我使用了一个代码从另一个表创建唯一值,在给定键的情况下获取我想要从表A中提取的行的引用。该表(表B)具有如下外观:

ID     TIME1        TIME2      
1      201502      201502    
2      201511      201511    

我已经设法通过执行一个简单的合并来取出我想要的值,该合并在给定引用的情况下从表A给出我想要的值。但是,我想使用“isin”函数来实现这一点。我根据下面的语法,它给了我重复的值。我唯一想要的是从表B中取出表中的行,参考表B.我如何设法做到这一点?

表C根据以下内容:

ID     TIME1       TIME2     VALUE_A      VALUE_B
1      201502      201502    a            1c
1      201502      201502    b            1d 
1      201502      201502    b            2e
1      201502      201502    b            6a

语法( “ISIN” -version):

subset = df[df.ID.isin(df2['ID']) & (df.TIME1.isin(df2['TIME1']) & df.TIME2.isin(df2['TIME2']))]

创建表A和表B的代码如下:


    df = DataFrame({'ID' : [1,1,1,1,1,1,1,1],
               'TIME1' : [201501,201502,201502,201501,201501,201501,201502,201502],
               'TIME2' : [201501,201502,201502,201501,201501,201501,201502,201502],
               'VALUE_A' : ['a', 'a', 'b', 'b', 'b', 'b', 'b', 'b'],
               'VALUE_B' : ['1a', '1c', '1d', '2e', '6a', '1d', '2e', '6a']})


    df2 = DataFrame({'ID' : [1,2],
                'TIME1' : [201502,201501],
                'TIME2' : [201502,201501]
                })

非常感谢提前!

2 个答案:

答案 0 :(得分:3)

我相信你想修改你的布尔条件:

In [146]:
subset = df[df.ID.isin(df2['ID']) & (df.TIME1.isin(df2['TIME1']) | df.TIME2.isin(df2['TIME2'])) ]
subset

Out[146]:
   ID   TIME1   TIME2 VALUE_A VALUE_B
1   1  201502  201-02       a      1c
2   1  201502  201502       b      1d
6   2  201511  201511       b      2e
7   2  201511  201511       b      6a

因此,这会检查ID是否存在以及Time1或Time2是否在另一个df中。

答案 1 :(得分:2)

只需使用{/ 1}}

即可实现此目的
isin()