将数据帧列表传递到lm()并查看结果 [英] Passing list of data frames into lm() and viewing the results
问题描述
我有三个数据帧,dfLON,dfMOS和dfATA.每个都具有相同的变量:y 是连续变量,a、b 和 c 是二元分类变量,还有一些 NA
.
I've got three data frames, dfLON, dfMOS, and dfATA. Each features the same variables: y is a continuous variable, and a, b, and c are binary categorical, and there is also some NA
.
我想建立单独的线性回归模型,每个数据集一个.
I'd like to build separate linear regression models, one for each data set.
使用我当前的代码,我设法制作了一个数据帧列表,并将其传递给lm().但是,是否有比 fitdfLON<-DfList [[1]]
等更简洁的查看结果的方法?在此示例中,我提供了三个数据帧,但实际上有〜25个,因此我不得不键入25次!
With my current code I've managed to make a list of data frames and pass it into lm(). But is there are more concise way to view the results than eg fitdfLON <- DfList[[1]]
? I've provided three data frames in this example but I actually have ~25 so I'd have to type it 25 times!
任何帮助将不胜感激.
起点(dfs):
dfLON <- data.frame(y=c(1.23,2.32,3.21,2.43),a=c(1,NA,1,2),b=c(1,1,2,2),c=c(2,1,2,1))
dfMOS <- data.frame(y=c(4.56,6.54,4.43,5.78),a=c(2,1,2,1),b=c(2,1,1,2),c=c(1,2,1,2))
dfATA <- data.frame(y=c(1.22,6.54,3.23,4.23),a=c(2,2,2,1),b=c(1,2,1,2),c=c(1,NA,1,2))
当前代码:
Mylm <- function(df){
fit <- lm(y ~ a + b + c, data=df)
return(fit)
}
DfList <- lapply(list(dfLON, dfMOS, dfATA), Mylm)
fitdfLON <- DfList[[1]]
fitdfMOS <- DfList[[2]]
fitdfATA <- DfList[[3]]
推荐答案
无论何时在许多不同的数据集上运行模型,都应使用扫帚库对其进行整理.这样会为每个模型生成一个干净的数据框,然后可以将其输出或用于下游分析中.
Whenever you're running models on many different datasets, it makes sense to tidy them using the broom library. This produces a clean data frame for each model, which you can then output or use in downstream analyses.
最简单的例子:
library(broom)
Mylm <- function(df){
fit <- lm(y ~ a + b + c, data=df)
tidy(fit) # tidy the fit object
}
list(dfLON, dfMOS, dfATA) %>% lapply(Mylm)
#[[1]]
# term estimate std.error statistic p.value
#1 (Intercept) 0.03 NaN NaN NaN
#2 a -0.78 NaN NaN NaN
#3 b 1.98 NaN NaN NaN
#
#[[2]]
# term estimate std.error statistic p.value
#1 (Intercept) 8.2975 0.969855 8.5554025 0.07407531
#2 a -1.6650 0.445000 -3.7415730 0.16626155
#3 b -0.3150 0.445000 -0.7078652 0.60785169
#
#[[3]]
# term estimate std.error statistic p.value
#1 (Intercept) 6.235 3.015000 2.067993 0.2867398
#2 a -2.005 1.740711 -1.151828 0.4551559
现在,您可以将其与purrr的 map_dfr()
函数结合起来,以将所有内容合并到一个合并的数据框中:
And you can now combine this with the map_dfr()
function from purrr to combine everything into one combined data frame:
library(purrr)
# note the named list entries; these will go into the "model" column
# without them, you'd just get a model number
list("LON" = dfLON, "MOS" = dfMOS, "ATA" = dfATA) %>%
map_dfr(Mylm, .id = "model")
# model term estimate std.error statistic p.value
#1 LON (Intercept) 0.0300 NaN NaN NaN
#2 LON a -0.7800 NaN NaN NaN
#3 LON b 1.9800 NaN NaN NaN
#4 MOS (Intercept) 8.2975 0.969855 8.5554025 0.07407531
#5 MOS a -1.6650 0.445000 -3.7415730 0.16626155
#6 MOS b -0.3150 0.445000 -0.7078652 0.60785169
#7 ATA (Intercept) 6.2350 3.015000 2.0679934 0.28673976
#8 ATA a -2.0050 1.740711 -1.1518281 0.45515586
为了使事情更紧凑,您可以在 map_dfr
中动态定义函数.当您要做的是拟合线性模型时,似乎很合适.
And to make things more compact, you can define the function on the fly inside map_dfr
. Seems appropriate when all you're doing is fit a linear model.
list("LON" = dfLON, "MOS" = dfMOS, "ATA" = dfATA) %>%
map_dfr(~ tidy(lm(y ~ a + b + c, data = .)),
.id = "model")
# model term estimate std.error statistic p.value
#1 LON (Intercept) 0.0300 NaN NaN NaN
#2 LON a -0.7800 NaN NaN NaN
#3 LON b 1.9800 NaN NaN NaN
#4 MOS (Intercept) 8.2975 0.969855 8.5554025 0.07407531
#5 MOS a -1.6650 0.445000 -3.7415730 0.16626155
#6 MOS b -0.3150 0.445000 -0.7078652 0.60785169
#7 ATA (Intercept) 6.2350 3.015000 2.0679934 0.28673976
#8 ATA a -2.0050 1.740711 -1.1518281 0.45515586
这篇关于将数据帧列表传递到lm()并查看结果的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持IT屋!