從R中的列匯總特定數據

Question

我正在嘗試在r中創建一個函數，該函數將讓我將最后三行的數據求和到一個新列中（即，如果我正在看第4天=第3 + 2 + 1天）

到目前為止，這是我已經完成的工作，但是行不通。

S3<- function(x){
res <- numeric(nrow(x))
for (i in 1:nrow(x)){    
res[i] <- i
if (i > 3) {             
  res[i] <- x[i-3,10]
} else {
  res[i] <- x[i,10]
}
}
x$PP3 <- res
return(x)
}

Answer 1

用這個：

x$PP3 <- x[,10] + c(0,head(x[,10],-1)) + c(0,0,head(x[,10],-2))

如果需要功能：

S3 <- function(x, j=10){
    within(x, PP3 <- x[,j] + c(0,head(x[,j],-1)) + c(0,0,head(x[,j],-2)))
}