R：循环逻辑以计算 30 天的移动总和

Question

我有几个帐户，我正在尝试计算按帐户分组的 30 天移动存款总和。我想做这个计算的 31 次迭代。因此，每个账户将有 31 个总和，代表 30 天的总存款金额。

例如，第一次迭代是从 10/1 到 10/30 的存款，第二次迭代是 10/2 到 10/31....然后最终是 11/1 到 11/30，例如。

我的第一个想法是做两个循环，按帐户和计算分组。

这里是一些示例数据，两个账户，10 和 11。第 1 天到第 61 天的存款金额。

x <- data.frame(day = c(1:61),
                account = ifelse(runif(61) <.5, 10, 11),
                amount = rnorm(61,mean = 100, sd = 10))

numinterations<-31
numsumdays<-30

monthsums<- for(i in 1:ndays){
              for(i in 1:numsumdays){
                 sum(testloop$daily_total) group by account
              }  


}

Answer 1

您可以使用zoo 中的rollapply 将移动窗口应用于列。根据要求，我避免使用dplyr。相反，数据框被拆分为数据框列表，并在最后重新组合。

我不得不先在这里填写缺少的日期/帐户组合，但它才能工作；可能有更优雅的方式来做到这一点。

library(zoo)

x <- data.frame(day = c(1:61),
            account = ifelse(runif(61) <.5, 10, 11),
            amount = rnorm(61,mean = 100, sd = 10))

all_combinations <- expand.grid(day=unique(x$day),account=unique(x$account))
x <- merge(x, all_combinations, all=TRUE)
x[is.na(x)] <- 0

lx <- split(x, x$account)
for (account in names(lx)) {
  lx[[account]][['rollingSum']] <- rollapply(lx[[account]][['amount']], width=30, FUN=sum, partial=TRUE, align='right')
}
names(lx) <- NULL
x <- do.call(rbind, lx)