Question

我想在多列数据表上使用rollapply函数，即我希望能够独立使用每一列，例如让我们考虑以下数据：

> DT = data.table(x=rep(c("a","b","c"),each=3), y=c(1,3,6), v=1:9)
> DT
   x y v
1: a 1 1
2: a 3 2
3: a 6 3
4: b 1 4
5: b 3 5
6: b 6 6
7: c 1 7
8: c 3 8
9: c 6 9

然后我想使用rollapply作为滚动子集，以便计算第2列和第3列的3个元素的滚动均值并将它们存储到外部变量中：

> r1= NA; r2 = NA
> ft=function(x) { r1=mean(x[,2,with=FALSE]) ; r2=mean(x[,3,with=FALSE]) }
> rollapply(DT, width=3, ft)
 Error in x[, 2, with = FALSE] : incorrect number of dimensions

除非我收到这个不方便的错误，为什么它不起作用？

输出结果为：

> r1
[1] 3.333333 3.333333 3.333333 3.333333 3.333333 3.333333 3.333333
> r2
[1] 2 3 4 5 6 7 8

Answer 1

仅使用Changed

添加其他选项

data.table

定义滚动平均函数

library(data.table) # v1.9.6+

在指定rollMean <- function(x, n) Reduce(`+`, shift(x, 0L:(n - 1L)))/n

时将其应用于多个列

.SDcols

Answer 2

你几乎可以做到：

lapply(DT[,2:3,with=F], function(x) rollapply(x,width=3, FUN=mean))
#$y
#[1] 3.333333 3.333333 3.333333 3.333333 3.333333 3.333333 3.333333

#$v
#[1] 2 3 4 5 6 7 8

如何在多列数据表上进行折叠

2 个答案: