создать накопительный счет, пока 2 из 6 предыдущих строк не удовлетворяют условию

Question

Если столбец а равен 1, я хотел бы начать накопительную сумму. Я хотел бы остановиться, когда 2 из предыдущих 6 строк равны 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 строк, затем использовать 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