Rolling Sum Dplyr答案

【问题标题】：Rolling Sum DplyrRolling Sum Dplyr
【发布时间】：2019-01-25 21:18:53
【问题描述】：

如果我有一个数据框，并且我想使用滚动求和来对前 n 行和后 n 行求和，那么最好的方法是什么？我知道roll_sum，但我找不到一种方法来使用它来适应我的用例。例如，假设我有一个向量 n。我指定我希望我的窗口为 1，这意味着对于每条记录，我想将它和两条相邻的记录相加。

n   window1
1   NA
3   8
4   12
5   15
6   18
7   22
9   17
1   15
5   6

如果我指定 2 作为我的窗口大小，那么结果如下：

n   window1 window2
1   NA  NA
3   8   NA
4   12  19
5   15  25
6   18  31
7   22  28
9   17  28
1   15  22
5   6   15

有没有简单的方法可以做到这一点？

【问题讨论】：

查看RcppRoll 包；你有roll_sum 函数。
我看到了，但我不确定使用它获得所需的输出。
相关Consecutive/Rolling sums in a vector in R
或Rolling sum in dplyr

标签： r dplyr rolling-sum

【解决方案1】：

可能有专用功能，但这似乎可行。它使您可以控制您希望它的行为方式。例如，前导函数中的default = 0 允许它转到最后一条记录，即使没有前导值。我敢打赌，这是相对缓慢且低效的。

library(dplyr)
library(purrr)

rolling_sum <- function(v, window = 1) {

  k <- 1:window

  vLag <- k %>%
    map_dfc(~lag(v, .))

  vLead <- k %>%
    map_dfc(~lead(v, ., default = 0))

  rowSums(bind_cols(vLag, V = v, vLead))

}

df <- data.frame(n = c(1,3,4,5,6,7,9,1,5))

df %>%
  mutate(window1 = rolling_sum(n, 1),
         window2 = rolling_sum(n, 2))

【讨论】：

【解决方案2】：

我认为 zoo 包中的 rollapplyr 是你的朋友。使用 align = ‘center’ 您可以将前 n 行和后 n 行相加。

【讨论】：