相关疑难解决方法(0)

来自R帮助do()的稍微改变的例子:

by_cyl <- group_by(mtcars, cyl)
models <- by_cyl %>% do(mod = lm(mpg ~ disp, data = .))
coefficients<-models %>% do(data.frame(coef = coef(.$mod)[[1]]))

在数据帧系数中,每个cyl组都有第一个线性模型系数.我的问题是如何生成一个数据框,其中不仅包含具有系数的列,还包含具有分组变量的列.

=====编辑:我扩展示例以尝试更清楚我的问题

让我们假设我想提取模型的系数和一些预测.我可以做这个:

by_cyl <- group_by(mtcars, cyl)
getpars <- function(df){
  fit <- lm(mpg ~ disp, data = df)
  data.frame(intercept=coef(fit)[1],slope=coef(fit)[2])
}
getprediction <- function(df){
  fit <- lm(mpg ~ disp, data = df)
  x <- df$disp
  y <- predict(fit, data.frame(disp= x), type = "response")
  data.frame(x,y)
}
pars <- by_cyl %>% do(getpars(.))
prediction <- by_cyl %>% do(getprediction(.)) …