按组回归

Question

只是一个非常简单的问题，我想使用 MASS 运行回归。 因变量分别为 val1、val2、val3，自变量为 a、b、c、d。

看看假数据就知道了。

library(data.table)
library(MASS)
test <- data.table(val1 = 1:10, val2 = 11:20, val3 = 21:30, a = rnorm(10), b = rnorm(10), c = rnorm(10), d = rnorm(10))
summary1 <- glm.nb(val1 ~ a + b + c + d, data = test)
summary2 <- glm.nb(val2 ~ a + b + c + d, data = test)
summary3 <- glm.nb(val3 ~ a + b + c + d, data = test)

我认为代码很丑陋。 我试过这个

for (i in c("val1", "val2", "val3")){
paste("sum_", c("val1", "val2", "val3"), sep = "") <- glm.nb(i ~ a + b + c + d, data = simple)
}

但它没有用。 关于改进的任何建议？ 原始数据中，大约有26个自变量，我觉得如果代码像这样sum1 <- glm.nb(val3 ~ a + b + c + d + e + f+ g + h + i + j + k + l, data = test)

我知道以下代码可能会有所帮助，但我不知道如何使用它们...:(

diff <- setdiff(colnames(test),c('val1','val2','val3'))

另外，我想知道lapply函数是否可以在data.table中实现这一点？

非常感谢！

Answer 1

最好将您的数据放在长格式中：

library(plyr)
library(reshape2)
xx <- melt(test,measure.vars=paste0('val',1:3))
ddply(xx,.(variable),function(x){
  coef(glm.nb(value~.,data=subset(x,select=-variable)))
})

 variable (Intercept)            a            b           c          d
1     val1    1.583602 -0.045909060 -0.018189342 0.026293033 0.29708648
2     val2    2.704601 -0.014641683 -0.003836401 0.006711503 0.10445377
3     val3    3.217729 -0.008925782 -0.001863267 0.003475509 0.06292286

如果您想要所有模型而不仅仅是系数：

dlply(xx,.(variable),function(x){
  glm.nb(value~.,data=subset(x,select=-variable))
})

Answer 2

使用您的循环方法，我只需将所有模型存储在这样的列表中

results <- list()

for (i in c("val1", "val2", "val3")){
  frml <- paste(i, "~ a + b + c + d")
  frml <- as.formula(frml)

  results[[i]] <- glm.nb(frml, data = simple)
}

然后通过查看results$val1等访问列表中的模型。

Answer 3

这是lapply的解决方案：

summary.list<-lapply(test[, .SD, .SDcols=patterns('val')],
                     function(i) glm.nb(i ~ a + b + c + d, data = test))