首页 文章

在R中的data.table环境中创建一个公式

提问于
浏览
19

我想在 data.table 内进行回归 . formula 需要动态构建 . 我尝试了以下方法:

x = data.table(a=1:20, b=20:1, id=1:5)
> x[,as.list(coef(lm(as.formula("a ~ b")))),by=id]
  Error in eval(expr, envir, enclos) : object 'a' not found

如何将环境指定为进行评估的实际data.table的环境?

编辑:我意识到我可以做lm(a~b) . 我需要公式是动态的,所以它被构建为一个字符串 . 通过动态我的意思是公式可以 paste0(var_1, "~", var_2) 其中 var_1 = avar_2 = b

这是一个解决方案,我认为我们可以做得更好:

txt = parse(text="as.list(coef(lm(a ~ b)))")
> x[,eval(txt),by=id]
  id (Intercept)  b
  1:  1          21 -1
  2:  2          21 -1
  3:  3          21 -1
  4:  4          21 -1
  5:  5          21 -1

1 回答

  • 15

    lm 可以接受一个字符串作为公式,所以将其与 .SD 组合如下:

    > x[, as.list(coef(lm("a ~ b", .SD))), by = id]
       id (Intercept)  b
    1:  1          21 -1
    2:  2          21 -1
    3:  3          21 -1
    4:  4          21 -1
    5:  5          21 -1
    

相关问题