2

我可以根据 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 中处理这个问题？谢谢。

4

1 回答 1

3

由于看起来每个步骤的结果都取决于您现在执行的方式中的前一个步骤，因此摆脱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]
于 2013-01-18T07:51:25.080 回答