r - 根据R中的前一行和当前行按组计算

Question

我可以根据 R 中的前一行和当前行进行计算。

对于此数据框(df):

    A B 
[1] 2 2 
[2] 2 3 
[3] 4 5.5 
... 

B2 = A2 + 0.5*B1 

我可以使用这段代码来计算这个函数。

for( i in 2:nrow(df) ) { 
        B[i] <- 1/2 * B[i-1] + df$A[i] 
} 

但是，我的数据集中有一个 Id(group)，所以我需要通过 id 来计算这个函数。

    id A B
[1] 1  0 0
[2] 1  2 2 
[3] 1  2 3
[4] 2  1 1 
[5] 2  4 4.5
[6] 2  4 6.25 
... 

B2 = A2 + 0.5*B1
B4 = A4 + 0.5*0 (becasue id is changed)
B5 = A5 + 0.5*B4 
... 

如何在 R 中处理这个问题？ 谢谢。

Answer 1

由于每一步的结果似乎都取决于您现在执行的方式中的前一步，因此摆脱 for 循环似乎很棘手。我想知道这样的事情是否足够快:

require(data.table)
require(compiler)

df <- data.table(df)

f <- function(A, B) {
  for(i in seq_along(B)[-1]) {
    B[i] <- 0.5 * B[i - 1L] + A[i]
  }

  B
}

g <- cmpfun(f)

df[, B := g(A, B), by = id]