在第一次出现某个值后删除行

Question

我有以下df：

df <- data.frame(var1 = c(1, 2, 2, 3, 4, 5, 5, 6, 7, 8, 9),
                 var2 = c("a", "b", "c", "d", "e", "f", "g", "h", "i", "j", "k"))

我想在达到前 5 个 (var1) 之后进行过滤。 所以结果应该是：

  var1 var2
1    1    a
2    2    b
3    2    c
4    3    d
5    4    e
6    5    f

我试过了：

   df1 <- df %>%
     mutate(a = !duplicated(var1)) %>%
     filter(var1 < 6 & a == TRUE)

但这里的问题是同时删除了 var2(c) 的重复项。

Answer 1

我们可以使用cumsum创建一个逻辑vector

df %>% 
  filter(cumsum(var1 == 5) < 2)
#    var1 var2
#1    1    a
#2    2    b
#3    2    c
#4    3    d
#5    4    e
#6    5    f

---

或者另一个选项是slice 和match

df %>%
   slice(seq(match(5, var1)))

---

或者which.max（假设有5个值）

df %>%
   slice(seq(which.max(var1 == 5)))