从data.table groupby模型中恢复lm()结果

发布于10月12日

我需要分析数据的一系列lm()输出的输出.表GROUP_BY回归:

library(data.table)
x <- c(1:5,66:70,101:110) 
y <- 31:50
g <- c( 1,1,1,1,1,1,1,1,1,1          
       ,2,2,2,2,2,2,2,2,2,2)

dt <- data.table(x,y,g)
mod <- dt[ , .(model = .(lm(x~y, .SD)))
           , by = g]

mod每组都有正确的型号g.现在，我想用残差向量 for each 组g创建一列:

mod[, resi := residuals(model)
    , by = g]

它似乎运行正常，但没有创建nem列resi:

> mod
       g    model
   <num>   <list>
1:     1 <lm[12]>
2:     2 <lm[12]>

为什么这不管用呢？

推荐答案

问题是您将残差()应用于list(model)，而不是直接应用于模型对象.

mod[, resi := .(list(residuals(model[[1]])))
    , by = g]

# Or 

mod[, resi := lapply(model, residuals)]

#        g    model                                                                                    resi
#    <num>   <list>                                                                                  <list>
# 1:     1 <lm[12]>                    10.909091,  1.818182, -7.272727,-16.363636,-25.454545, 25.454545,...
# 2:     2 <lm[12]> -1.471465e-14, 6.482041e-15, 5.472051e-15, 4.101520e-15, 2.842011e-15, 1.526991e-15,...