r - 在r中的group_by之后建模后取消嵌套列表列

我想对group_by之后的所有组进行线性回归，将模型系数保存在列表列中，然后使用“unnest”来“扩展列表列”。
这里我以mtcars数据集为例。

注意:我想使用do' here, because broom::tidy`不适用于所有模型。

mtcars %>% group_by(cyl) %>%
    do(model=lm(mpg~wt+hp, data=.)) %>%
    mutate(coefs = list(summary(model)$coefficients)) %>%
    unnest()

我想要这样的东西。

cyl   term         Estimate Std. Error   t value     Pr(>|t|)
 4     (Intercept) 36.9083305 2.19079864 16.846975 1.620660e-16
 4     wt         -2.2646936 0.57588924 -3.932516 4.803752e-04
 4     hp          -0.0191217 0.01500073 -1.274718 2.125285e-01
 6.......
 6......
........

我收到如下错误:

Error: All nested columns must have the same number of elements.

谁能帮助解决这个问题？尝试了很多次后，我无法理解它。

最佳答案

一种选择是提取'coefs'列(.$coefs)，用'cyl'列设置list列的名称，用list遍历map，将其转换为data.frame，根据行名创建一个新列，然后使用.id从names的list创建“cyl”列

library(tidyverse)
mtcars %>%
   group_by(cyl) %>%
   do(model=lm(mpg~ wt + hp, data=.)) %>%
   mutate(coefs = list(summary(model)$coefficients)) %>%
                 select(-model) %>%
   {set_names(.$coefs, .$cyl)} %>%
   map_df(~ .x %>%
               as.data.frame %>%
               rownames_to_column('term'), .id = 'cyl')
# cyl        term    Estimate Std. Error   t value     Pr(>|t|)
#1   4 (Intercept) 45.83607319 4.78693568  9.575243 1.172558e-05
#2   4          wt -5.11506233 1.60247105 -3.191984 1.276524e-02
#3   4          hp -0.09052672 0.04359827 -2.076383 7.151610e-02
#4   6 (Intercept) 32.56630096 5.57482132  5.841676 4.281411e-03
#5   6          wt -3.24294031 1.37365306 -2.360815 7.759393e-02
#6   6          hp -0.02219994 0.02017664 -1.100279 3.329754e-01
#7   8 (Intercept) 26.66393686 3.66217797  7.280896 1.580743e-05
#8   8          wt -2.17626765 0.72094143 -3.018647 1.168393e-02
#9   8          hp -0.01367295 0.01073989 -1.273099 2.292303e-01

如果我们想使用tidy，则将map_df的内容更改为

       ...                %>%
        map_df(~ .x %>%
                          broom::tidy(.), .id = 'cyl')

另外，另一个选择是在nest之后添加group_by，然后将broom::tidy应用于model对象，然后再将unnest应用于

mtcars %>%
   group_by(cyl) %>%
   nest %>%
   mutate(data = map(data, ~ .x %>%
                    summarise(model = list(broom::tidy(lm(mpg ~ wt + hp)))))) %>%
   unnest %>%
   unnest

关于r - 在r中的group_by之后建模后取消嵌套列表列，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/54455273/