Estrarre elementi dall'elenco annidato utilizzando solo le funzioni del pacchetto purrr

Question

Come estrarre gli elementi da un elenco nidificato utilizzando solo il pacchetto purrr? In questo caso voglio ottenere un vettore di intercettazioni dopo aver diviso un data.frame. Ho realizzato ciò di cui ho bisogno usando lapply(), ma mi piacerebbe usare solo le funzioni del pacchetto purrr.

library(purrr) 
mtcars %>% 
split(.$cyl) %>% 
map( ~lm(mpg ~ wt, data = .)) %>%  # shorthand NOTE: ~ lm 
lapply(function(x) x[[1]] [1]) %>% # extract intercepts <==is there a purrr function for this line? 
as_vector()        # convert to vector 

Ho cercato map() e at_depth(), ma nulla sembrava funzionare per me.

Answer 1

Le funzioni map dispongono di una codifica stenografica per l'indicizzazione di elenchi annidati. Un utile snippet dalla pagina di aiuto:

Per indicizzare in profondità in un elenco nidificato, utilizzare più valori; c ("x", "y") è equivalente a z [["x"]] [["y"]].

Quindi, utilizzando il codice per gli indici annidati insieme map_dbl, che si riduce a un vettore, si può semplicemente fare:

mtcars %>% 
    split(.$cyl) %>% 
    map(~lm(mpg ~ wt, data = .)) %>% 
    map_dbl(c(1, 1)) 

     4  6  8 
39.57120 28.40884 23.86803 

Ho trovato anche questo blog post introducendo purrr 0.1.0 utile, in quanto ha dato un qualche altro esempio della codifica stenografica che ho finito per usare.