r - 具有潜力的累计和 "resets"

Question

我有一个数字向量，我想为其计算一种累积和。我说“一种”是因为标准的 cumsum 基本上是 cumsum[i] = cumsum[i-1] + x[i]。但是，在这种情况下，我需要使用 cumsum[i] = max(cumsum[i-1] + x[i], x[i])。

也就是说，如果最近的元素的值大于累积和(例如，如果之前有负值，就会发生这种情况)，则直接采用最近的元素。

这显然可以通过简单的 for 循环手动完成:

set.seed(1)
x <- runif(10, min = -1, max = 1)
csum <- rep(0, 10)
for(i in seq_along(x)) {
  if (i == 1) csum[i] <- x[i]

  csum[i] <- max(csum[i-1] + x[i], x[i])
}
x; csum
#>  [1] -0.4689827 -0.2557522  0.1457067  0.8164156 -0.5966361  0.7967794
#>  [7]  0.8893505  0.3215956  0.2582281 -0.8764275
#>  [1] -0.4689827 -0.2557522  0.1457067  0.9621223  0.3654862  1.1622655
#>  [7]  2.0516161  2.3732117  2.6314397  1.7550123

^{由 reprex package 创建于 2020-04-27 (v0.3.0)}

但是有没有一种方法可以避免 for 循环呢？我一直在努力想出一个办法，但就是想不出来。

如果相关，我的真实案例会将其应用于数据框。它将被分组，然后我将为每个组创建一个包含此累积总和的新列。但我对那部分很满意，只是想不出如何清理这个操作。

Answer 1

我们可以在base R中使用Reduce

csum2 <- Reduce(function(u, v) max(u + v, v), x, accumulate = TRUE)

-检查OP的输出

identical(csum, csum2)
#[1] TRUE

---

或者另一种选择是 accumulate from purrr

library(purrr)
accumulate(x, ~  max(.x + .y, .y))