将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)

我想为x的子集创建一个单独的lm,并且y共享相同的ID.以下代码完成了工作:

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)和未矢量化.我还想将所有lms存储在单个数据结构中.必须有一种优雅的方式来做到这一点,但我找不到它.有帮助吗？

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

使用plyr包中的一些魔法.该函数dlply采用a 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 

Answer 3

怎么样

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

？