是否有像rollapply(标准R或CRAN包中)这样的函数在data.frame上运行,但不会将其转换为矩阵。 rollapply可以与data.frame一起使用,但如果data.frame具有混合类型,则每个数据窗口都将转换为字符(矩阵)。
我更喜欢支持width,na.pad,align等的函数,就像rollapply
一样示例
使用混合类型的任何data.frame
test = data.frame( Name = c( "bob" , "jane" , "joe" ) , Points = c( 4 , 9 , 1 ) )
假设你想要使用窗口大小2进行滚动。使用data.frame调用FUN的第一次迭代,其中只包含test的第1行和第2行。
首次迭代时RollapplyThatRespectsDataFrame( ... , FUN = function( x ) { ... } )
设置x = data.frame( Name = c( "bob" , "jane" ) , Points = c( 4 , 9 ) )
第二次迭代是一个data.frame,其中包含test
的第2行和第3行。
基本上这个新函数与rollapply做同样的事情,除了它与data.frames一起正常工作。它不会转换为矩阵。
答案 0 :(得分:8)
试试这个:
> library(zoo)
> DF <- data.frame(a = 1:10, b = 21:30, c = letters[1:10])
> replace(DF, 1:2, rollapply(DF[1:2], 3, sum, fill = NA))
a b c
1 NA NA a
2 6 66 b
3 9 69 c
4 12 72 d
5 15 75 e
6 18 78 f
7 21 81 g
8 24 84 h
9 27 87 i
10 NA NA j
关于在讨论之后添加到问题中的示例,通过将此功能应用于行索引,可以在rollapply
之上分层这些功能:
> lapply(as.data.frame(t(rollapply(1:nrow(test), 2, c))), function(ix)test[ix, ])
$V1
Name Points
1 bob 4
2 jane 9
$V2
Name Points
2 jane 9
3 joe 1
在这里它的包装更好一点:
rollapply.data.frame <- function(data, ..., fill = NULL, FUN,
simplify = function(x) do.call(rbind, x)) {
fill0 <- if (!is.null(fill)) NA
result <- lapply(
as.data.frame(t(rollapply(1:nrow(data), ..., fill = fill0, FUN = c))),
function(ix) {if (all(is.na(ix))) fill else FUN(data[ix, ])}
)
simplify(result)
}
> rollapply(test, 2, FUN = identity, simplify = identity)
$V1
Name Points
a bob 4
b jane 9
$V2
Name Points
b jane 9
c joe 1
> rollapply(test, 2, FUN = identity, fill = NA, simplify = identity)
$V1
Name Points
a bob 4
b jane 9
$V2
Name Points
b jane 9
c joe 1
$V3
[1] NA