第二次使用 dplyr::do 时丢失 group_by 信息

Question

我运行在我的数据集的多个部分上使用多个模型，类似于（但有更多模型）

library(tidyverse)
d1 <- mtcars %>% 
  group_by(cyl) %>% 
  do(mod_linear = lm(mpg ~ disp + hp, data = ., x = TRUE))
d1 
# Source: local data frame [3 x 3]
# Groups: <by row>
# 
# # A tibble: 3 x 3
#     cyl mod_linear 
# * <dbl> <list>     
# 1    4. <S3: lm>   
# 2    6. <S3: lm>   
# 3    8. <S3: lm>

然后我整理这个 tibble 并使用 broom 包中的 tidy() 保存我的参数估计。

我还想计算预测变量的标准差（在我设置 x = TRUE 时存储在上面的模型中）以创建然后比较 re-scaled 参数。我可以使用

做前者

d1 %>%
  # group_by(cyl) %>% 
  do(term = colnames(.$mod$x),
     pred_sd = apply(X = .$mod$x, MARGIN = 2, FUN = sd)) %>%
  unnest()
# # A tibble: 9 x 2
#          term  pred_sd
#         <chr>    <dbl>
# 1 (Intercept)  0.00000
# 2        disp 26.87159
# 3          hp 20.93453
# 4 (Intercept)  0.00000
# 5        disp 41.56246
# 6          hp 24.26049
# 7 (Intercept)  0.00000
# 8        disp 67.77132
# 9          hp 50.97689

但是，结果不是分组的小标题，所以我最终失去了 cyl 列来告诉我哪些术语属于哪个模型。如何避免这种损失？ - 再次添加 group_by 似乎会引发错误。

n.b。我想至少在第一部分（拟合模型）避免使用 purrr，因为我运行不同类型的模型然后需要重塑结果（d1），我喜欢进度条do。

n.b。我想使用模型的 $x 组件而不是原始数据，因为它们具有正确比例的数据（我正在试验预测变量的不同转换）

Answer 1

我们可以先 nesting 然后 unnest

mtcars %>% 
    group_by(cyl) %>% 
      nest(-cyl) %>% 
      mutate(mod_linear = map(data, ~ lm(mpg ~ disp + hp, data = .x, x = TRUE)),
           term  = map(mod_linear, ~ names(coef(.x))),
           pred = map(mod_linear, ~ .x$x %>%
                                       as_tibble %>% 
                                       summarise_all(sd) %>% 
                                       unlist )) %>%
   select(-data, -mod_linear) %>%
     unnest
# A tibble: 9 x 3
#    cyl term         pred
#  <dbl> <chr>       <dbl>
#1  6.00 (Intercept)   0  
#2  6.00 disp         41.6
#3  6.00 hp           24.3
#4  4.00 (Intercept)   0  
#5  4.00 disp         26.9
#6  4.00 hp           20.9
#7  8.00 (Intercept)   0  
#8  8.00 disp         67.8
#9  8.00 hp           51.0

或者不是多次调用 map，而是可以使用

进一步压缩

mtcars %>% 
       group_by(cyl) %>% 
        nest(-cyl) %>% 
        mutate(mod_contents = map(data, ~ {
           mod <- lm(mpg ~ disp + hp, data = .x, x = TRUE)
           term <- names(coef(mod))
           pred <- mod$x %>%
                      as_tibble %>%
                            summarise_all(sd) %>%
                            unlist
            tibble(term, pred)        
            }
         )) %>%
      select(-data) %>%
      unnest    
# A tibble: 9 x 3
#    cyl term         pred
#  <dbl> <chr>       <dbl>
#1  6.00 (Intercept)   0  
#2  6.00 disp         41.6
#3  6.00 hp           24.3
#4  4.00 (Intercept)   0  
#5  4.00 disp         26.9
#6  4.00 hp           20.9
#7  8.00 (Intercept)   0  
#8  8.00 disp         67.8
#9  8.00 hp           51.0

如果我们从 'd1' 开始（基于 OP 的代码）

d1 %>% 
   ungroup %>%
   mutate(mod_contents = map(mod_linear, ~ {

             pred <- .x$x %>%
                       as_tibble %>%
                        summarise_all(sd) %>%
                        unlist
            term <- .x %>%
                          coef %>% 
                          names 
            tibble(term, pred)                        

        }))  %>%
   select(-mod_linear) %>%
   unnest

第二次使用 dplyr::do 时丢失 group_by 信息

Loosing group_by information when using dplyr::do for the second time

r

dplyr

purrr