创建一个累积计数，直到前6行中的2行满足条件为止

Question

如果列a等于1，我想开始一个累加的总和。 我想在前6行中的2行等于0时停止。

dplyr::tibble(a = c(1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 0, 1, 1, 1), 
          sum = c(1, 2, 3, 4, 5, 6, 7, 8, 8, 9, 0, 1, 2, 3))

总和是我想要的输出

理想情况下使用tidyverse

Answer 1

一种方法是找出在6行的间隔内找到两个连续0的行，然后使用cumsum创建组并最终在每个组中取cumsum值。

library(dplyr)
library(purrr)

df %>%
   mutate(sum1 = map_dbl(seq_along(a), ~sum(a[. : max(.-6, 1)] == 0) >= 2)) %>%
   group_by(group = cumsum(sum1 != lag(sum1, default = first(sum1)))) %>%
   mutate(ans = cumsum(a)) %>%
   ungroup %>%
   select(-sum1, -group)

# A tibble: 14 x 2
#      a   ans
#   <dbl> <dbl>
# 1     1     1
# 2     1     2
# 3     1     3
# 4     1     4
# 5     1     5
# 6     1     6
# 7     1     7
# 8     1     8
# 9     0     8
#10     1     9
#11     0     0
#12     1     1
#13     1     2
#14     1     3