Question

根据我学到的this威胁，下面data.table中变量b的滚动总和可以实现如下：

数据创建+计算滚动总和：

    x <- data.table(a = sample(letters[1:3], 100, replace = TRUE), b = runif(100))
    setorder(x, a)

    # alternative 1
    x[, .(b, Reduce(`+`, shift(b, 0:2))), by = a]
    # alternative 2
    x[, .(b, stats::filter(b, rep(1, 3), sides = 1)), by = a]

当前+期望的输出：

     a           b        V2   V2_desired
  1: a 0.457665568        NA   0.457665568
  2: a 0.752555834        NA   1.210221
  3: a 0.864672124 2.0748935   2.0748935
  4: a 0.542168656 2.1593966   2.1593966
  5: a 0.197962875 1.6048037   1.6048037

现在为前两个障碍物生成了NA。在每个小组中。我需要调整其中一个替代方案，只计算当前的阻值。在组指数开始的情况下（位于第2位）的情况（最后两个障碍）。这应该是可推广的，这样我可以考虑最后n个值的窗口并处理异常。

有什么想法吗？

Answer 1

我并非100％确定我得到了您所需要的内容，但默认情况下shift函数会留下NA值。您可以通过传递fill参数来更改该行为。在您的情况下，由于您要对数据进行求和，因此您可能希望使用fill=0进行尝试：

set.seed( 123 )
x[, .(b, Reduce(`+`, shift(b, 0:2, fill=0))), by = a]

head返回：

   a         b       V2
1: a 0.5999890 0.599989
2: a 0.8903502 1.490339
3: a 0.7205963 2.210935
4: a 0.5492847 2.160231
5: a 0.9540912 2.223972
6: a 0.5854834 2.088859

使用data.table在组内的最后n个观测值的滚动总和中避免NA

1 个答案: