拟合多个LHS的线性模型

我是R新手,我想用*apply函数来改进下面的脚本(我已经阅读了关于apply ,但是我无法设法使用它)。 我想在多个独立variables(它们是数据框中的列)上使用lm函数。 我用了

 for (i in (1:3) { assign(paste0('lm.',names(data[i])), lm(formula=formula(i),data=data)) } 

Formula(i)被定义为

 formula=function(x) { as.formula ( paste(names(data[x]),'~', paste0(names(data[-1:-3]), collapse = '+')), env=parent.frame() ) } 

谢谢。

如果我没有弄错你,你正在使用这样的数据集:

 set.seed(0) dat <- data.frame(y1 = rnorm(30), y2 = rnorm(30), y3 = rnorm(30), x1 = rnorm(30), x2 = rnorm(30), x3 = rnorm(30)) 

x1x2x3是协variables,而y1y2y3是三个独立的响应。 您正试图拟合三个线性模型:

 y1 ~ x1 + x2 + x3 y2 ~ x1 + x2 + x3 y3 ~ x1 + x2 + x3 

目前,您正在使用y1y2y3循环,每次只适合一个模型。 而且你希望通过用lapplyreplacefor循环来加速这个过程。

我想指出,你是在错误的轨道上。 lm()是一个昂贵的操作。 只要你的数据集不小, for循环的成本是可以忽略的。 用lapplyreplace循环不会提高性能。

由于三种模型都具有相同的RHS,所以三种模型的模型matrix是相同的。 因此,所有模型的QR分解只需要进行一次。 lm允许这个,你可以使用:

 fit <- lm(cbind(y1, y2, y3) ~ x1 + x2 + x3, data = dat) #Coefficients: # y1 y2 y3 #(Intercept) -0.081155 0.042049 0.007261 #x1 -0.037556 0.181407 -0.070109 #x2 -0.334067 0.223742 0.015100 #x3 0.057861 -0.075975 -0.099762 

如果你检查str(fit) ,你会发现这不是三个线性模型的列表; 相反,它是一个具有单个$qr对象的单一线性模型,但具有多个LHS。 所以$coefficients$residuals$fitted.values$fitted.values是matrix。

如果你有更多的协variables,你可以避免使用键入或粘贴公式.

 fit <- lm(cbind(y1, y2, y3) ~ ., data = dat) #Coefficients: # y1 y2 y3 #(Intercept) -0.081155 0.042049 0.007261 #x1 -0.037556 0.181407 -0.070109 #x2 -0.334067 0.223742 0.015100 #x3 0.057861 -0.075975 -0.099762 

小心:不要写:

 y1 + y2 + y3 ~ x1 + x2 + x3 

这将把y = y1 + y2 + y3当作一个单一的回应。 使用cbind()


跟进:

感谢您的答复。 我对概括感兴趣。 让我更具体一点。 我有一个dataframedf ,其中前n列是自variables(y1,y2,y3,....) ,接下来的m列是因variables(x1+x2+x3+....) 。 对于n=3m=3fit <- lm(cbind(y1, y2, y3) ~ ., data = dat)) 。 但是,如何通过使用df的结构自动完成。 我的意思是类似于(for i in (1:n)) fit <- lm(cbind(df[something] ~ df[something], data = dat)) 。 那个“东西”我用pastepaste0创build了它。 谢谢。

您仍然可以通过pastepaste来创build模型公式:

 paste0("cbind(", paste(names(df)[1:n], collapse = ", "), ")", " ~ .") 

例如,使用iris数据集:

 paste0("cbind(", paste(names(iris)[1:2], collapse = ", "), ")", " ~ .") # "cbind(Sepal.Length, Sepal.Width) ~ ." 

您可以将此string公式传递给lm ,因为lm将自动将其强制转换为公式类。