基于组内的对进行过滤

Question

 Group Code
   1     2
   1     2
   1     4
   1     1
   2     4 
   2     1
   2     2 
   2     3
   2     1
   2     1
   2     3

在每个组中都有对。例如在第 1 组中；对是 (2,2),(2,4),(4,1)

我想根据出现在对开头的代码号 2 过滤这些对。 例如在第 1 组中，只有 (2,2) 和 (2,4) 将被保留，而 (4,1) 将被过滤掉。

异常输出：

 Group Code
   1     2
   1     2
   1     4
   2     2 
   2     3
  

Answer 1

检查groupbyshift

out = df[df.groupby('Group').Code.apply(lambda x : x.eq(2) | x.eq(2).shift())]
Out[67]: 
   Group  Code
0      1     2
1      1     2
2      1     4
6      2     2
7      2     3