根据另一列中的任何先前值更改列

Question

我有一个像这样的数据框：

foo <- c(rep(FALSE, 5), TRUE, rep(FALSE, 4))
rank_order <- seq(11,20)

df <- data.frame(rank_order = as.numeric(rank_order),
                 foo = foo)

我想做的是在rank_order行之后的每个df$foo == TRUE值中加一个。这意味着rank_order应该看起来像这样：

rank_order_target <- c(11, 12, 13, 14, 15, 17, 18, 19, 20, 21)

更改rank_order的一个值很容易，而lag会查看foo的一个先前值（如下所示），但是如何查看{ {1}}？

foo

可以使用df %>% mutate(rank_order_new = case_when(lag(foo, default = FALSE) == TRUE ~ rank_order + 1, TRUE ~ rank_order)) rank_order foo rank_order_new 1 11 FALSE 11 2 12 FALSE 12 3 13 FALSE 13 4 14 FALSE 14 5 15 FALSE 15 6 16 TRUE 16 7 17 FALSE 18 8 18 FALSE 18 9 19 FALSE 19 10 20 FALSE 20 解决方案，也可以使用base这样的解决方案。

Answer 1

我们可以使用cumsum

library(dplyr)
df %>%
     mutate(new = rank_order + cumsum(foo))