有没有
我在R中运行滚动回归，使用存储在data.table的数据。
我有一个工作版本，但它感觉像一个黑客 - 我真的使用我从zoo包中知道的，并没有data.table中的魔法......因此，它感觉比它应该慢。
结合约书亚的建议（见下文），通过使用lm.fit而不是lm来加速约12倍。
（修改）示例代码：
require(zoo)
require(data.table)
require(rbenchmark)
set.seed(1)
tt <- seq(as.Date("2011-01-01"), as.Date("2012-01-01"), by="day")
px <- rnorm(366, 95, 1)
DT <- data.table(period=tt, pvec=px)
dtt <- DT[,tnum:=as.numeric(period)][, list(pvec, tnum)]
dtx <- as.matrix(DT[,tnum:=as.numeric(period)][, tnum2:= tnum^2][, int:=1][, list(pvec, int, tnum, tnum2)])
rollreg <- function(dd) coef(lm(pvec ~ tnum + I(tnum^2), data=as.data.frame(dd)))
rollreg.fit <- function(dd) coef(lm.fit(y=dd[,1], x=dd[,-1]))
rr <- function(dd) rollapplyr(dd, width=20, FUN = rollreg, by.column=FALSE)
rr.fit <- function(dd) rollapplyr(dd, width=20, FUN = rollreg.fit, by.column=FALSE)
bmk <- benchmark(rr(dtt), rr.fit(dtx),
columns = c('test', 'elapsed', 'relative'),
replications = 10,
order = 'elapsed'
)
test elapsed relative
2 rr.fit(dtx) 0.48 1.0000
1 rr(dtt) 5.85 12.1875
试着应用这里和这里显示的知识，我制作了以下简单的滚动回归函数，我认为它使用了data.table操作的一些速度。
请注意，问题有点不同（并且更加现实）：采用矢量，添加时滞，并对其本身进行倒退。 这类AR类问题相当广泛。
我在这里分享它，因为它可能是有用的，我相信它可以改进（我会更新，因为我改进）：
require(data.table)
set.seed(1)
x <- rnorm(1000)
DT <- data.table(x)
DTin <- data.table(x)
lagDT <- function(DTin, varname, l=5)
{
i = 0
while ( i < l){
expr <- parse(text =
paste0(varname, '_L', (i+1),
':= c(rep(NA, (1+i)),', varname, '[-((length(', varname, ') - i):length(', varname, '))])'
)
)
DTin[, eval(expr)]
i <- i + 1
}
return(DTin)
}
rollRegDT <- function(DTin, varname, k=20, l=5)
{
adj <- k + l - 1
.x <- 1:(nrow(DTin)-adj)
DTin[, int:=1]
dtReg <- function(dd) coef(lm.fit(y=dd[-c(1:l),1], x=dd[-c(1:l),-1]))
eleNum <- nrow(DTin)*(l+1)
outMatx <- matrix(rep(NA, eleNum), ncol = (l+1))
colnames(outMatx) <- c('intercept', 'L1', 'L2', 'L3', 'L4', 'L5')
for (i in .x){
dt_m <- as.matrix(lagDT(DTin[i:(i+adj), ], varname, l))
outMatx[(i+(adj)),] <- dtReg(dt_m)
}
return(outMatx)
}
rollCoef <- rollRegDT(DT, varname='x')
据我所知，并不是。 data.table没有任何滚动窗口的特殊功能。 其他软件包已经在矢量上实现了滚动功能，所以它们可以用在data.table的j中。 如果它们效率不够高，并且没有软件包具有更快的版本（？），那么你自己编写更快的版本，并且（当然）贡献它们：既可以是现有软件包，也可以创建自己的软件包。
相关问题（请点击链接中的链接）：
使用data.table来加速rollapply
R data.table滑动窗口
R中的多列滚动回归
你可以正确加快你的rollreg.fit与我在这里发布的答案相当多。
上一篇: Is there a