大熊猫isin功能

rey 发表于 Dev

雷

我有一个数据框：

In [47]: df
Out[47]: 
    uid   a   b
0   111   1   2
1   111   2   3
2   111   4   5
3   111   6   7
4   111   5   8
5   222   0   9
6   222  11  12
7   222  13  11
8   222   2   1
9   333  14  16
10  333   3   2
11  333  16  19
12  333   3   4
13  444  21  20
14  444   9   5
15  444  20  24
16  444   5   6

我只想检查inb中的值是否存在a，反之亦然uid。我用过isin：

df[(df.b.isin(df.a))|(df.a.isin(df.b))]

但这会遍历所有值，并且无法提供所需的输出。

所需的输出：

 Out[49]: 
    uid   a   b
0   111   1   2
1   111   2   3
2   111   4   5
4   111   5   8
6   222  11  12
7   222  13  11
9   333  14  16
11  333  16  19
13  444  21  20
14  444   9   5
15  444  20  24
16  444   5   6

耶斯列尔

我认为您需要groupby并申请boolean indexing：

print (df.groupby('uid').apply(lambda x: x[(x.b.isin(x.a))|(x.a.isin(x.b))]))
        uid   a   b
uid                
111 0   111   1   2
    1   111   2   3
    2   111   4   5
    4   111   5   8
222 6   222  11  12
    7   222  13  11
333 9   333  14  16
    11  333  16  19
444 13  444  21  20
    14  444   9   5
    15  444  20  24
    16  444   5   6

print (df.groupby('uid')
         .apply(lambda x: x[(x.b.isin(x.a))|(x.a.isin(x.b))])
         .reset_index(drop=True))
    uid   a   b
0   111   1   2
1   111   2   3
2   111   4   5
3   111   5   8
4   222  11  12
5   222  13  11
6   333  14  16
7   333  16  19
8   444  21  20
9   444   9   5
10  444  20  24
11  444   5   6