R - data.table 中两列的滚动总和答案

【问题标题】：R - Rolling sum of two columns in data.tableR - data.table 中两列的滚动总和
【发布时间】：2021-04-21 01:58:08
【问题描述】：

我有一个如下的data.table -

dt = data.table(
  date = seq(as.Date("2015-12-01"), as.Date("2015-12-10"), by="days"),
  v1 = c(seq(1, 9), 20),
  v2 = c(5, rep(NA, 9))
)
dt
          date v1 v2
 1: 2015-12-01  1  5
 2: 2015-12-02  2 NA
 3: 2015-12-03  3 NA
 4: 2015-12-04  4 NA
 5: 2015-12-05  5 NA
 6: 2015-12-06  6 NA
 7: 2015-12-07  7 NA
 8: 2015-12-08  8 NA
 9: 2015-12-09  9 NA
10: 2015-12-10 20 NA

问题 1：我想将 v1 的当前行值与 v2 的前一行值相加，因此输出如下所示。

          date v1 v2
 1: 2015-12-01  1  5
 2: 2015-12-02  2  7
 3: 2015-12-03  3 10
 4: 2015-12-04  4 14
 5: 2015-12-05  5 19
 6: 2015-12-06  6 25
 7: 2015-12-07  7 32
 8: 2015-12-08  8 40
 9: 2015-12-09  9 49
10: 2015-12-10 20 69

我曾尝试使用 rollapplyr 函数来实现，但失败了。

问题 2：我想将函数 qma 滚动应用到 v1 的当前行值和 v2 的前一行值，而不是添加（如问题 1） qma <- function(x, y){(x+y+7)/2}

我确信必须有一种简单的方法可以使用 data.table 在一行中执行此操作。

谢谢

【问题讨论】：

当然。我为这个后续问题创建了另一个帖子 - stackoverflow.com/questions/65746358/…

标签： r data.table rolling-computation rollapply

【解决方案1】：

您可以将first v2 值添加到v1 的累积总和中。

library(data.table)
dt[, v2:= first(v2) + c(0, cumsum(v1[-1]))]
dt

#          date v1 v2
# 1: 2015-12-01  1  5
# 2: 2015-12-02  2  7
# 3: 2015-12-03  3 10
# 4: 2015-12-04  4 14
# 5: 2015-12-05  5 19
# 6: 2015-12-06  6 25
# 7: 2015-12-07  7 32
# 8: 2015-12-08  8 40
# 9: 2015-12-09  9 49
#10: 2015-12-10 10 59

【讨论】：

上面的数据只是作为一个简单的例子。实际上，v1 和 v2 中的数字可以是随机的。我对 data.table 做了一些改动。你能更新你的解决方案吗？
@Saurabh 您是否尝试过我对更新数据的回答？它仍然对我有用，并给出了如图所示的预期输出。