计算以零为底的累积和（cumsum）答案

【问题标题】：Calculate cumulative sum (cumsum) floored at zero计算以零为底的累积和（cumsum）
【发布时间】：2020-08-28 14:46:23
【问题描述】：

我想修改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

【问题讨论】：

因为 16+10-5=21
你能帮帮我吗？

标签： r dataframe cumsum

【解决方案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 之前还是之后......

【讨论】：

【解决方案2】：

您可以通过accumulate()从purrr自定义累积和规则。

library(dplyr)
library(purrr)

TD %>%
  group_by(product) %>% 
  mutate(result = accumulate(D-S, ~ pmax(0, .x + .y), .init = 0)[-1])

# # 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 的第一个元素为负数，它就不起作用 - pmax 只开始作用于第二行。跨度>
@pseudospin 感谢您的提醒！现在如果D-S 的第一个元素是负数，它就可以工作了。
由于accumulate 迭代地处理每个元素，因此无需使用pmax。使用max 将在此处产生相同的结果。