r - 如果组具有特定值(查看前一个值)，则保留组的行

Question

对于每个学生id，我想检查他们在第一次失败后是否通过了测试。

library(dplyr)
data.frame(id = c(1,1,1,2,2,3,4,4), homework = c('fail', 'fail', 'pass', 'fail', 'pass', 'pass', 'pass', 'pass'), 
           date = c('20200122', '20200124', '20200126', '20210214', '20210216', '20210303', '20201203', '20201205')) %>%
  mutate(date = as.Date(date, '%Y%m%d'))

# id homework       date
# 1  1     fail 2020-01-22
# 2  1     fail 2020-01-24
# 3  1     pass 2020-01-26
# 4  2     fail 2021-02-14
# 5  2     pass 2021-02-16
# 6  3     pass 2021-03-03
# 7  4     pass 2020-12-03
# 8  4     pass 2020-12-05

如果是这样，我想保留该学生的行:

# id homework       date
# 1  1     fail 2020-01-22
# 2  1     fail 2020-01-24
# 3  1     pass 2020-01-26
# 4  2     fail 2021-02-14
# 5  2     pass 2021-02-16

我似乎无法找到如何“到达”组内上方/下方的行以进行检查。有人有什么建议吗？预先感谢:)

Answer 1

您可以使用 dplyr::lag 和 dplyr::lead 分别访问当前行之前和之后的行。

df %>%
  group_by(id) %>%
  mutate(
    passed_after_fail = homework == "pass" & lag(homework) == "fail"
  ) %>%
  filter(any(passed_after_fail))