dplyr 和 tidyr - 使用因子一次性计算大量线性模型

在阅读了 tidyverse 的更多内容后，我开始立即拟合许多线性模型，如中所述this https://blog.rstudio.org/2016/02/02/tidyr-0-4-0/。也就是说，我会按照以下方式做一些事情：

library(dplyr)
library(tidyr)
library(purrr)
df <- data.frame(y = rnorm(10), 
                 x1 = runif(10),
                 x2 = runif(10))

df %>%
  gather(covariate, value, x1:x2) %>% 
  group_by(covariate) %>% 
  nest() %>% 
  mutate(model = map(.x = data , .f = ~lm(y ~ value, data = .))) %>% 
  mutate(rsquared = map_dbl(.x = model, .f = ~summary(.)$r.squared))

问题是，当变量类型不同时，例如当一个是数字而一个是因子时，这种方法会失败，因为gather()函数将强制整体value向量化为因子。例如，

df <- data.frame(y = rnorm(10), 
                 x1 = runif(10),
                 x3 = sample(c("a", "b", "c"), 10, replace = TRUE))

df %>%
  gather(covariate, value, x1:x3) %>% 
  sapply(class)

随后是警告

Warning message:
attributes are not identical across measure variables; they will be dropped 

          y   covariate       value 
  "numeric" "character" "character" 

and the value列是一个字符，所以技巧nest()将不再起作用，因为所有协变量都将作为因子放入。

我想知道是否有tidy这样做的方法。

您可以在拟合模型时转换类型，但您应该按照注释中指出的那样小心行事，因为这可能会产生意想不到的后果。

如果您仍然想转换，您可以使用type_convert from readr在整个框架上或type.convert就在“值”向量上。

Using type_convert:

mutate(model = map(.x = data , .f = ~lm(y ~ value, data = readr::type_convert(.))))

Using type.convert:

mutate(model = map(.x = data , .f = ~lm(y ~ type.convert(value), data = .)))

作为链的一部分，其中任何一个都会导致本例所需的结果：

df %>%
    gather(covariate, value, x1:x3) %>% 
    group_by(covariate) %>% 
    nest() %>% 
    mutate(model = map(.x = data , .f = ~lm(y ~ type.convert(value), data = .))) %>% 
    mutate(rsquared = map_dbl(.x = model, .f = ~summary(.)$r.squared))

# A tibble: 2 x 4
  covariate              data    model   rsquared
      <chr>            <list>   <list>      <dbl>
1        x1 <tibble [10 x 2]> <S3: lm> 0.33176960
2        x3 <tibble [10 x 2]> <S3: lm> 0.06150498