dplyr / R cumulative sum with reset

by

I think you can use accumulate() here to help. And i’ve also made a wrapper function to use for different thresholds

sum_reset_at <- function(thresh) {
  function(x) {
    accumulate(x, ~if_else(.x>=thresh, .y, .x+.y))
  }  
}

tib %>% mutate(c = sum_reset_at(5)(a))
#       t     a     c
#   <dbl> <dbl> <dbl>
# 1     1     2     2
# 2     2     3     5
# 3     3     1     1
# 4     4     2     3
# 5     5     2     5
# 6     6     3     3
tib %>% mutate(c = sum_reset_at(4)(a))
#       t     a     c
#   <dbl> <dbl> <dbl>
# 1     1     2     2
# 2     2     3     5
# 3     3     1     1
# 4     4     2     3
# 5     5     2     5
# 6     6     3     3
tib %>% mutate(c = sum_reset_at(6)(a))
#       t     a     c
#   <dbl> <dbl> <dbl>
# 1     1     2     2
# 2     2     3     5
# 3     3     1     6
# 4     4     2     2
# 5     5     2     4
# 6     6     3     7

More Related Contents:

  1. Replace values greater than zero with 1 in r
  2. How does one access the piped data.frame in dplyr::funs?
  3. Extract row corresponding to minimum value of a variable by group
  4. How to select the rows with maximum values in each group with dplyr? [duplicate]
  5. Filter multiple values on a string column in dplyr
  6. Concatenate strings by group with dplyr [duplicate]
  7. Can dplyr package be used for conditional mutating?
  8. What does %>% function mean in R?
  9. Dplyr join on by=(a = b), where a and b are variables containing strings?
  10. R cumulative sum by condition with reset
  11. Proper idiom for adding zero count rows in tidyr/dplyr
  12. Why is enquo + !! preferable to substitute + eval
  13. Add NAs to make all list elements equal length
  14. Assign intermediate output to temp variable as part of dplyr pipeline
  15. Can I replace NAs when joining two data frames with dplyr?
  16. pass function arguments to both dplyr and ggplot
  17. Removing NA observations with dplyr::filter()
  18. Mutate multiple columns in a dataframe
  19. dplyr issues when using group_by(multiple variables)
  20. dplyr – using mutate() like rowmeans()
  21. dplyr piping data – difference between `.` and `.x`
  22. Filling missing dates in a grouped time series – a tidyverse-way?
  23. mutate_each / summarise_each in dplyr: how do I select certain columns and give new names to mutated columns?
  24. Spread with duplicate identifiers (using tidyverse and %>%) [duplicate]
  25. How to change the now deprecated dplyr::funs() which includes an ifelse argument?
  26. Apply a ggplot-function per group with dplyr and set title per group
  27. Mutate multiple variable to create multiple new variables
  28. How to pass column name as argument to function for dplyr verbs?
  29. How to use data.table within functions and loops?
  30. Override column types when importing data using readr::read_csv() when there are many columns

Leave a Comment