Extraire une colonne dplyr tbl comme vecteur

Existe-t-il une méthode plus succincte pour obtenir une colonne d’un tpl de dplyr en tant que vecteur, à partir d’un tbl avec le back-end de la firebase database (la trame / table de données ne peut pas être directement associée)?

require(dplyr) db <- src_sqlite(tempfile(), create = TRUE) iris2 <- copy_to(db, iris) iris2$Species # NULL 

Cela aurait été trop facile, alors

 collect(select(iris2, Species))[, 1] # [1] "setosa" "setosa" "setosa" "setosa" etc. 

Mais cela semble un peu maladroit.

Avec dplyr 0.7.0, vous pouvez utiliser pull pour obtenir un vecteur à partir d’un tbl .

 library("dplyr") #> #> Attaching package: 'dplyr' #> The following objects are masked from 'package:stats': #> #> filter, lag #> The following objects are masked from 'package:base': #> #> intersect, setdiff, setequal, union db <- src_sqlite(tempfile(), create = TRUE) iris2 <- copy_to(db, iris) vec <- pull(iris2, Species) head(vec) #> [1] "setosa" "setosa" "setosa" "setosa" "setosa" "setosa" 

Selon le commentaire de @nacnudus, il semble qu’une fonction pull été implémentée dans dplyr 0.6:

 iris2 %>% pull(Species) 

Pour les anciennes versions de dplyr, voici une fonction intéressante qui permet de tirer une colonne un peu mieux (plus facile à taper et plus facile à lire):

 pull <- function(x,y) {x[,if(is.name(substitute(y))) deparse(substitute(y)) else y, drop = FALSE][[1]]} 

Cela vous permet de faire l'un ou l'autre:

 iris2 %>% pull('Species') iris2 %>% pull(Species) iris2 %>% pull(5) 

Résultant en...

  [1] 21.0 21.0 22.8 21.4 18.7 18.1 14.3 24.4 22.8 19.2 17.8 16.4 17.3 15.2 10.4 10.4 14.7 32.4 30.4 33.9 21.5 15.5 15.2 13.3 19.2 27.3 26.0 30.4 15.8 19.7 15.0 21.4 

Et cela fonctionne aussi très bien avec les blocs de données:

 > mtcars %>% pull(5) [1] 3.90 3.90 3.85 3.08 3.15 2.76 3.21 3.69 3.92 3.92 3.92 3.07 3.07 3.07 2.93 3.00 3.23 4.08 4.93 4.22 3.70 2.76 3.15 3.73 3.08 4.08 4.43 [28] 3.77 4.22 3.62 3.54 4.11 

Une belle façon de le faire en v0.2 de dplyr :

 iris2 %>% select(Species) %>% collect %>% .[[5]] 

Ou si vous préférez:

 iris2 %>% select(Species) %>% collect %>% .[["Species"]] 

Ou si votre table n'est pas trop grande, simplement ...

 iris2 %>% collect %>% .[["Species"]] 

Vous pouvez également utiliser unlist plus facile à lire car vous n’avez pas besoin de répéter le nom de la colonne ou de spécifier l’index.

 iris2 %>% select(Species) %>% unlist(use.names = FALSE) 

J’utiliserais la fonction magrittr commodité de magrittr :

 library(magrittr) library(dplyr) iris2 %>% select(Species) %>% extract2(1) 

J’écrirais probablement:

 collect(select(iris2, Species))[[1]] 

Puisque dplyr est conçu pour travailler avec des fichiers de données, il n’y a pas de meilleur moyen d’obtenir une seule colonne de données.

@ Luke1018 a proposé cette solution dans un des commentaires:

Vous pouvez également utiliser l’opérateur d’exposition magrittr ( %$% ) pour extraire un vecteur d’un magrittr de données.

Par exemple:

 iris2 %>% select(Species) %>% collect() %$% Species 

Je pensais que cela méritait sa propre réponse.