跨两行两列保留满足条件的记录

Question

我有一个类似这样的数据集:

df <- 
  read.table(textConnection("ID   Column1   Column2
A 0 1
A 1 0
A 1 0
A 1 0
A 0 1
A 1 0
A 0 1
A 0 0 
A 1 0 
A 1 0
B 0 1
B 1 0 
C 0 1
C 0 0
C 1 0"), header=TRUE)

我想在 dplyr 中做一个 group_by ID，它维护 Column2 = '1' 的记录，并且它下面的记录有 Column1 = '1'。每个 ID 可能发生不止一次；应排除所有其他记录。所以上面的输出应该是:

<表类="s-表"><头>身份证第 1 列第 2 列<正文>一个01一个10一个01一个10B01B10

非常感谢任何帮助，谢谢!

Answer 1

您可以使用lag 和lead:

library(dplyr)

df %>%
  group_by(ID) %>%
  filter(lead(Column1) == 1 &     Column2  == 1 |
              Column1  == 1 & lag(Column2) == 1) %>%
  ungroup()

# # A tibble: 6 × 3
#   ID    Column1 Column2
#   <chr>   <int>   <int>
# 1 A           0       1
# 2 A           1       0
# 3 A           0       1
# 4 A           1       0
# 5 B           0       1
# 6 B           1       0