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（组），所以我需要通过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]

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

1 个答案: