将 lm 应用到由帧的第三列定义的数据帧的子集

Question

我有一个数据框，其中包含 x 值向量、y 值向量和 ID 向量：

x <- rep(0:3, 3)
y <- runif(12)
ID <- c(rep("a", 4), rep("b", 4), rep("c", 4))
df <- data.frame(ID=ID, x=x, y=y)

我想为共享相同 ID 的 x 和 y 子集创建一个单独的 lm。以下代码可以完成工作：

a.lm <- lm(x~y, data=subset(df, ID=="a"))
b.lm <- lm(x~y, data=subset(df, ID=="b"))
c.lm <- lm(x~y, data=subset(df, ID=="c"))

除了这非常脆弱（未来的数据集可能有不同的 ID）并且未矢量化。我还想将所有流媒体存储在一个数据结构中。一定有一种优雅的方法可以做到这一点，但我找不到。有什么帮助吗？

Answer 1

使用base函数，您可以split原始数据帧并对其使用lapply：

lapply(split(df,df$ID),function(d) lm(x~y,d))
$a

Call:
lm(formula = x ~ y, data = d)

Coefficients:
(Intercept)            y  
    -0.2334       2.8813  


$b

Call:
lm(formula = x ~ y, data = d)

Coefficients:
(Intercept)            y  
     0.7558       1.8279  


$c

Call:
lm(formula = x ~ y, data = d)

Coefficients:
(Intercept)            y  
      3.451       -7.628  

Answer 2

怎么样

library(nlme) ## OR library(lme4)
lmList(x~y|ID,data=d)

？

Answer 3

使用 plyr 包中的一些魔法。函数 dlply 接受一个 data.frame，将其拆分，对每个元素应用一个函数，然后将其组合到一个 list 中。这非常适合您的应用。

library(plyr)
#fitList <- dlply(df, .(ID), function(dat)lm(x~y, data=dat))
fitList <- dlply(df, .(ID), lm, formula=x~y) # Edit

这将创建一个列表，其中包含 ID 的每个子集的模型：

str(fitList, max.level=1)

List of 3
 $ a:List of 12
  ..- attr(*, "class")= chr "lm"
 $ b:List of 12
  ..- attr(*, "class")= chr "lm"
 $ c:List of 12
  ..- attr(*, "class")= chr "lm"
 - attr(*, "split_type")= chr "data.frame"
 - attr(*, "split_labels")='data.frame':    3 obs. of  1 variable:

这意味着您可以对列表进行子集化并使用它。例如，要获取 lm 模型的系数，其中 ID=="a"：

> coef(fitList$a)
(Intercept)           y 
   3.071854   -3.440928