r - 计算以零为底的累积总和 (cumsum)

Question

我想修改 cumsum 函数。我想将负值更改为 0。并且当有一个不同于 0 的新 D 值时，则使用 D-S。

下面的例子:

TD <- data.frame(product = rep("A", 7),
                 data = seq(as.Date("2020-01-01"), as.Date("2020-01-07"), by = "day"),
                 D = c(74, 0, 0, 0, 0, 20, 10), S = c(20, 30, 20, 5, 2, 4, 5))

TD <- TD %>% group_by(product) %>%  mutate(result = cumsum(D) - cumsum(S))

> TD
# A tibble: 6 x 5
# Groups:   product [1]
  product data           D     S result      I expected
  <chr>   <date>     <dbl> <dbl>  <dbl>
1 A       2020-01-01    74    20     54          54
2 A       2020-01-02     0    30     24          24
3 A       2020-01-03     0    20      4           4
4 A       2020-01-04     0     5     -1           0
5 A       2020-01-05     0     2     -3           0
6 A       2020-01-06    20     4     13          16
7 A       2020-01-07    10     5     18          21

Answer 1

我认为这个功能可以满足您的需求

pos_cumsum <- function(x) {
  cs <- cumsum(x)
  cm <- cummin(cs)
  return (cs - pmin(cm, 0))
}

TD<- TD%>% group_by(product) %>%  mutate(result = pos_cumsum(D-S))
TD
#> # A tibble: 7 x 5
#> # Groups:   product [1]
#>   product data           D     S result
#>   <chr>   <date>     <dbl> <dbl>  <dbl>
#> 1 A       2020-01-01    74    20     54
#> 2 A       2020-01-02     0    30     24
#> 3 A       2020-01-03     0    20      4
#> 4 A       2020-01-04     0     5      0
#> 5 A       2020-01-05     0     2      0
#> 6 A       2020-01-06    20     4     16
#> 7 A       2020-01-07    10     5     21

虽然我想知道 D 是发生在 S 之前还是之后......