将 `zoo:rollapply` 的 `width` 参数作为参数传递给被调用函数

Question

如何将zoo:rollapply 函数的width 参数传递给rollapply 内调用的FUN 函数？

我将值和宽度作为相同data.frame 的列：

library(zoo)
a = data.frame(v = 1:10, w = c(2,1,3,5,2,7,3,2,1,3))
#     v w
# 1   1 2
# 2   2 1
# 3   3 3
# 4   4 5
# 5   5 2
# 6   6 7
# 7   7 3
# 8   8 2
# 9   9 1
# 10 10 3

我能做到：

rollapply(a$v, a$w, function(x) sum(x), partial=T)

这给出了：

[1] 3 2 9 20 11 42 21 17 9 19

现在我想在计算中使用a$v 的每个滚动窗口a$w 的对应值。例如：

rollapply(a$v, a$w, function(x) sum(x) + a$w[1], partial=T)

但不是将a$w[1] 作为一个常数值（这里，它只是将上面的值加 2），我想使用每次对应于a$w 的a$w 中的值（即，在a 的同一行）。

因此期望的输出是：

[1] 5 3 12 25 13 49 24 19 10 22

Answer 1

1) 宽度不必在函数中。之后可以添加：

rollapply(a$v, a$w, sum, partial = TRUE) + a$w
## [1]  5  3 12 25 13 49 24 19 10 22

如果您想要不同的对齐方式，请指定 align=。

2)这种方式有点难看但是另一种方式是维护一个外部索引。

i <- 0
rollapply(a$v, a$w, function(x) sum(x) + a$w[i <<- i+1], partial = TRUE)
## [1]  5  3 12 25 13 49 24 19 10 22

2a) 这可以通过使用面向对象的思想来维护状态以牺牲额外代码为代价进行清理。这里我们定义了一个原型对象p，它有一个内部计数器和一个incr 方法，该方法每次在对象上调用incr 时递增并返回它。

library(proto)
p <- proto(counter = 0, incr = function(.) .$counter <- .$counter + 1)
rollapply(a$v, a$w, function(x) sum(x) + a$w[p$incr()], partial = TRUE)
## [1]  5  3 12 25 13 49 24 19 10 22

3) 问题中的示例使用居中对齐，但如果您确实需要右对齐或左对齐，则可以通过迭代 a 而不是仅仅在 a$v 上完成。这里是右对齐。

Sum <- function(x) {
  x <- matrix(x,,2)
  v <- x[, 1]
  w <- tail(x[, 2], 1)
  sum(v) + w
}
rollapplyr(a, a$w, Sum, partial = TRUE, by.column = FALSE)
## [1]  3  3  9 15 11 28 21 17 10 30

# double check
rollapplyr(a$v, a$w, sum, partial = TRUE) + a$w
## [1]  3  3  9 15 11 28 21 17 10 30