提取模型摘要並將它們存儲爲一個新列

Question

我是purrr模式的新手，並且一直在努力。

以下幾個來源，我設法嵌套數據框，在嵌套數據上運行線性模型，從每個lm中提取一些係數，併爲每個lm生成一個摘要。我想要做的最後一件事是從摘要中提取「r.squared」（我認爲這將是我試圖實現的最簡單的部分），但無論出於何種原因，我無法獲得語法對。

這裏是什麼，我有一個MWE的作品：

library(purrr) 
library(dplyr) 
library(tidyr) 

mtcars %>% 
    nest(-cyl) %>% 
    mutate(fit = map(data, ~lm(mpg ~ wt, data = .)), 
     sum = map(fit, ~summary)) 

，這裏是我的嘗試以提取失敗r.squared：

mtcars %>% 
    nest(-cyl) %>% 
    mutate(fit = map(data, ~lm(mpg ~ wt, data = .)), 
     sum = map(fit, ~summary), 
     rsq = map_dbl(sum, "r.squared")) 

Error in eval(substitute(expr), envir, enclos) : 
    `x` must be a vector (not a closure) 

這是表面上類似於RStudio網站上給出的示例：

mtcars %>% 
    split(.$cyl) %>% 
    map(~ lm(mpg ~ wt, data = .x)) %>% 
    map(summary) %>% 
    map_dbl("r.squared") 

這個工作，但我想r.squared值坐在一個新的列（因此mutate語句），我想了解爲什麼我的代碼不工作，而不是解決問題的工作。

編輯：

下面是我來使用下面的解決方案的工作方案：

mtcars %>% 
     nest(-cyl) %>% 
     mutate(fit = map(data, ~lm(mpg ~ wt, data = .)), 
      summary = map(fit, glance), 
      r_sq = map_dbl(summary, "r.squared")) 

編輯2：

所以，它實際上原來，錯誤是由包含在summary = map（fit，〜summary）行中的波浪號鍵。我的猜測是使對象成爲嵌套的函數，而不是摘要本身返回的對象。 。喜歡一個權威的答案就這一點，如果有人想插話

需要明確的是，這個版本的原代碼的正常工作：

mtcars %>% 
    nest(-cyl) %>% 
    mutate(fit = map(data, ~lm(mpg ~ wt, data = .)), 
     summary = map(fit, summary), 
     r_sq = map_dbl(summary, "r.squared")) 

Answer 1

，以適應當前的管道，你會想要使用map和glance沿着unnest從broom包。

library(tidyr) 
library(dplyr) 
library(broom) 

mtcars %>% 
    nest(-cyl) %>% 
    mutate(fit = map(data, ~lm(mpg ~ wt, data = .))) %>% 
    unnest(map(fit, glance)) 

你得到的不僅僅是R平方以上，並從那裏你可以使用select砸你不需要的東西。

如果你想保持嵌套在列表列模型摘要：

mtcars %>% 
    nest(-cyl) %>% 
    mutate(fit = map(data, ~lm(mpg ~ wt, data = .)), 
     summary = map(fit, glance)) 

如果你想只提取你只需要使用map實際值嵌套框架單個值（和不是[[或extract2，正如我原先所建議的，非常感謝你的發現）。

mtcars %>% 
    nest(-cyl) %>% 
    mutate(fit = map(data, ~lm(mpg ~ wt, data = .)), 
     summary = map(fit, glance), 
     r_sq = map_dbl(summary, "r.squared")) 

Answer 2

我想你想達到什麼樣的，你是關閉使用glance()功能從broom包好：

library(broom) 
library(dplyr) 
mtcars %>% 
    group_by(cyl) %>% 
    do(glance(lm(mpg ~ wt, data = .))) %>% 
    select(cyl, r.squared) 
# cyl r.squared 
# <dbl>  <dbl> 
#1  4 0.5086326 
#2  6 0.4645102 
#3  8 0.4229655 

Answer 3

必須有一個更好的辦法，這是我嘗試用管：

mtcars %>% 
    split(.$cyl) %>% 
    map(~ lm(mpg ~ wt, data = .x)) %>% 
    map(summary) %>% 
    map_dbl("r.squared") %>% 
    list() %>% 
    as.data.frame(col.names = "r.squared") %>% 
    add_rownames(var = "cyl") 

# # A tibble: 3 × 2 
#  cyl r.squared 
# <chr>  <dbl> 
# 1  4 0.5086326 
# 2  6 0.4645102 
# 3  8 0.4229655 

注意：您可能會收到警告。

警告消息：棄用，請改用tibble :: rownames_to_column（）。