通过在R中的数据帧的行上应用Reduce函数来创建新列

Question

我有一个包含ID，日期和观察到的回报的数据框。它可以比作：

df <- data.frame(
  ID = gl(3, 10, labels = c("A", "B", "C")), 
  Date = factor(rep(2006, 2015, 3)), 
  lr = runif(30, -0.01, 0.01))

现在我想使用以下函数来查找每个ID的指数移动平均值的向量，并将它们作为新列添加到我的原始数据帧中：

Emean<-function(x){
    ema <- function(a,b) {lambda*a+(1-lambda)*b}
    Reduce(ema, x, accumulate=T)
}

所以我希望结果数据框有列ID，Date，lr和mlr。最后一列（mlr）将使用上述函数计算;并且（对不起用于表示错误！）但这是公式：

mlr_t=lambda*mlr_t-1 + (1-lambda)*lr_t

＆＃39; _t＆＃39;表示时间。

现在正如我所说，我想将我的函数应用于按ID分组的行，并将结果作为列添加到此数据框中。 ＆＃39; Reduce＆＃39;的输出无法直接添加到该数据框中，我必须在几个步骤中对其进行操作，这在R中非常耗时。

我需要一个计算效率高的解决方案来完成我说的话。在实际数据集中，每个ID都有+ 100K ID和+250个日期。

Answer 1

作为

mlr_0 = 0
mlr_1 = 0 + (1-lambda)*lr_1
mlr_2 = lambda * mlr_1 + (1-lambda)*lr_2
      = lambda * (1-lambda) * lr_1 + (1-lambda)*lr_2
mlr_3 = lambda * mlr_2 + (1-lambda)*lr_3
      = lambda^2 * (1-lambda) * lr_1 + lambda * (1-lambda) * lr_2 + (1-lambda)*lr_3
...
mlr_t = lambda^(t-1) * (1-lambda) * lr_1 + lambda^(t-2) * (1-lambda) * lr_2 + ...
      = \Sum_{i=1}^{t} lambda^(t-i) * (1-lambda)*lr_i

你可以这样做（使用data.table）

setDT(df)
lambda <- 0.5
# This calculates the lambda^(t-i)
l <- function(i, lambda){ lambda^(i-seq_len(i)) }

# This calculates multiplies element wise and sums up the mlr_3
my_fun <- function(x, lr, lambda){
  sum((1-lambda) * c(0,lr)[1:x] * l(x, lambda))}

# Apply both function to the vector
df[, vapply(seq_len(.N), my_fun, numeric(1), lr, lambda)  ,by = ID]

结果（set.seed(42)）

    ID        V1
 1:  A 0.0000000
 2:  A 0.4574030
 3:  A 0.6972392
 4:  A 0.4916894
 5:  A 0.6610685
 6:  A 0.6514070
 7:  A 0.5852515
 8:  A 0.6609199
 9:  A 0.3977932
10:  A 0.5273928
11:  B 0.0000000
12:  B 0.2288709
...