一个类似于R中diff的迭代和滞后函数，但不仅仅是差？

Question

x = c(1, 2, 1, 3, 11, 7, 5)
diff(x)
# [1]  1 -1  2  8 -4 -2
diff(x, lag=2)
[1]  0  1 10  4 -6

有没有自定义这个，这样我们就可以使用不同的功能？例如，Sum：

itersum(x)
# 3 3 4 14 18 12

Answer 1

在基R中，有filter函数。它不像zoo::rollapply那样友好和通用，但速度非常快。在您的示例中，您希望应用具有权重的卷积过滤器C(1,1):

itersum <- function(x, n = 2) tail(filter(x, rep(1, n)), sides = 1), -(n-1))

itersum(x)
# 3 3 4 14 18 12

为了让您了解更多信息，下面是如何根据filter重写diff和cumsum函数：

diff   <- function(x) head(filter(x, c(1, -1)), -1)
cumsum <- function(x) filter(x, 1, method = "recursive")

通常，如果您要滚动二进制函数，那么head和tail可能是最简单和最快的方法，因为它将利用向量化函数：

itersum     <- function(x) tail(x, -1) + head(x, -1)
diff        <- function(x) tail(x, -1) - head(x, -1)
sign.change <- function(x) tail(sign(x), -1) != head(sign(x), -1)