Cumulative sum in R with reset after reaching threshold

cumsum_with_reset.R

testvector <- c(.5, .1, .2, .9, .9, .2, .5)

# group rows based on cumsum with reset
cumsum_group <- function(x, threshold) {
  cumsum <- 0
  group <- 1
  result <- numeric()
  
  for (i in 1:length(x)) {
    cumsum <- cumsum + x[i]
    
    if (cumsum > threshold) {
      group <- group + 1
      cumsum <- x[i]
    }
    
    result = c(result, group)
    
  }
  
  return (result)
}

# cumsum with reset
cumsum_with_reset <- function(x, threshold) {
  cumsum <- 0
  group <- 1
  result <- numeric()
  
  for (i in 1:length(x)) {
    cumsum <- cumsum + x[i]
    
    if (cumsum > threshold) {
      group <- group + 1
      cumsum <- x[i]
    }
    
    result = c(result, cumsum)
    
  }
  
  return (result)
}

# test
cumsum_group(testvector, 1)
cumsum_with_reset(testvector, 1)

# read_excel("c:\\Users/jgilf_000/Desktop/cumsum_test.xlsx") %>% 
#   group_by(var1) %>% 
#   mutate(group = cumsum_group(val2, 1),
#          cumsum = cumsum_with_reset(val2, 1)
#          )