从lm函数获取分组数据的p值

时间:2016-03-17 15:46:58

标签: r dplyr plyr lm

我正在尝试使用lm()函数和plyr包为我的数据中的每个细分拟合一个模型,因为我的数据是按密钥分组的。

我设法运行模型并获得系数以及R ^ 2& adj。平方但我正在努力解决p值。

library("plyr")
#Sample data
test_data <- data.frame(key = c("a","a","a","a","a","b","b","b","b","b"),
y = c(100,180,120,60,140,200,220,240,260,280),
x1 = c(50,60,79,85,90,133,140,120,160,170),
x2 =  c(20,18,47,16,15,25,30,25,20,15))

#model
model_1 <- dlply(test_data, .(key), 
    function(test_data) lm(y ~ x1 +     x2,data = test_data))

#coefficients
ldply(model_1, coef)

#adj r-squared
ldply(model_1, function(x) summary(x)$r.squared)

我已经尝试了这个,它获取了密钥和p值,但它没有我需要能够将输出与模型中的系数合并的变量名称

#p-values but missing the variable names
ldply(model_1, function(x) summary(x)$coefficients)[,c(1,5)]

我尝试使用Do然后使用tidy dplyr中的<div> <form name="myForm"> <!-- more html code --> </form> </div> 来适应模型,这可以很好地处理小数据集,因为它实际上返回了我需要的所有内容但是我的实际数据包含超过1,000个不同的段,并且RStudio最终崩溃。

2 个答案:

答案 0 :(得分:0)

我认为不需要plyrsapply会做得很好。

sapply(model_1, function(x) summary(x)$coefficients[, 4])

                    a         b
(Intercept) 0.3699423 0.3013515
x1          0.7698867 0.7307786
x2          0.9764913 0.3814288

t()将获得与您的估算值相同的配置。

顺便说一句,您可能希望查看multidplyr包,以及tidydplyr::do

答案 1 :(得分:0)

我正在使用“dplyr”包来格式化输出。在“dlply”函数中使用的函数中,你应该使用summary()来表示lm(),所以当你调用“coef”时它也会包含p.values。

test_data <- data.frame(key = c("a","a","a","a","a","b","b","b","b","b"),
                        y = c(100,180,120,60,140,200,220,240,260,280),
                        x1 = c(50,60,79,85,90,133,140,120,160,170),
                        x2 =  c(20,18,47,16,15,25,30,25,20,15))

model<-by(test_data,test_data$key,function(x)summary(lm(y~x1+x2,x)))

R2<-t(data.frame(lapply(model,function(x)x$adj.r.squared)));colnames(R2)<-"R2_adj";R2

      R2_adj
a -0.8939647
b  0.4292186

Co<-as.data.frame(t(data.frame(lapply(model,function(x)x$coef))))

colnames(Co)<-c("intercept","x1","x2")

library(dplyr)

Co%>%
        mutate(key=substr(rownames(Co),1,1),
               variable=substr(rownames(Co),3,12))%>%
        select(key,variable,intercept,x1,x2)

  key   variable   intercept         x1          x2
1   a   Estimate 162.1822438 -0.6037364  0.07628315
2   a Std..Error 141.3436897  1.8054132  2.29385395
3   a    t.value   1.1474318 -0.3344035  0.03325545
4   a   Pr...t..   0.3699423  0.7698867  0.97649134
5   b   Estimate 271.0532276  0.3624009 -3.62853907
6   b Std..Error 196.2769562  0.9166979  3.25911570
7   b    t.value   1.3809733  0.3953330 -1.11335080
8   b   Pr...t..   0.3013515  0.7307786  0.38142882