Wyodrębnij kolumnę dplyr tbl jako wektor

Czy istnieje bardziej zwięzły sposób, aby uzyskać jedną kolumnę dplyr tbl jako wektor, z tbl z zapleczem bazy danych (tzn. ramka/Tabela danych nie może być bezpośrednio podzbiorem)?

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

To byłoby zbyt łatwe, więc

collect(select(iris2, Species))[, 1]
# [1] "setosa"     "setosa"     "setosa"     "setosa"  etc.
Ale wydaje się trochę niezdarny.
Author: smci, 2014-02-07

6 answers

Z dplyr 0.7.0, możesz użyć pull, Aby uzyskać wektor z 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"
 93
Author: Lorenz Walthert,
Warning: date(): Invalid date.timezone value 'Europe/Kyiv', we selected the timezone 'UTC' for now. in /var/www/agent_stack/data/www/doraprojects.net/template/agent.layouts/content.php on line 54
2017-06-14 10:49:28

Zgodnie z komentarzem @nacnudus, wygląda na to, że pull funkcja została zaimplementowana w dplyr 0.6:

iris2 %>% pull(Species)

Dla starszych wersji dplyr, oto zgrabna funkcja, która sprawi, że wyciąganie kolumny będzie nieco przyjemniejsze (łatwiejsze do wpisania i łatwiejsze do odczytania):

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

To pozwala Ci zrobić jedno z tych:

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

W wyniku czego...

 [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

I działa również dobrze z ramkami danych:

> 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

Fajny sposób na to w v0.2 z dplyr:

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

Lub jeśli wolisz:

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

Lub jeśli twój stół nie jest zbyt duży, po prostu...

iris2 %>% collect %>% .[["Species"]]
 75
Author: Tommy O'Dell,
Warning: date(): Invalid date.timezone value 'Europe/Kyiv', we selected the timezone 'UTC' for now. in /var/www/agent_stack/data/www/doraprojects.net/template/agent.layouts/content.php on line 54
2017-12-05 17:48:27

Możesz również użyć unlist, które uważam za łatwiejsze do odczytania, ponieważ nie musisz powtarzać nazwy kolumny ani podawać indeksu.

iris2 %>% select(Species) %>% unlist(use.names = FALSE)
 58
Author: StanislawSwierc,
Warning: date(): Invalid date.timezone value 'Europe/Kyiv', we selected the timezone 'UTC' for now. in /var/www/agent_stack/data/www/doraprojects.net/template/agent.layouts/content.php on line 54
2015-06-27 10:02:05

Użyłbym extract2 funkcji wygody z magrittr:

library(magrittr)
library(dplyr)

iris2 %>%
  select(Species) %>%
  extract2(1)  
 19
Author: Hugh,
Warning: date(): Invalid date.timezone value 'Europe/Kyiv', we selected the timezone 'UTC' for now. in /var/www/agent_stack/data/www/doraprojects.net/template/agent.layouts/content.php on line 54
2014-11-26 09:20:05

Pewnie bym napisał:

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

Ponieważ dplyr jest przeznaczony do pracy z TBL danych, nie ma lepszego sposobu na uzyskanie pojedynczej kolumny danych.

 17
Author: hadley,
Warning: date(): Invalid date.timezone value 'Europe/Kyiv', we selected the timezone 'UTC' for now. in /var/www/agent_stack/data/www/doraprojects.net/template/agent.layouts/content.php on line 54
2014-02-07 13:33:19

@Luke1018 zaproponował takie rozwiązanie w jednym z komentarzy:

Można również użyć operatora ekspozycji magrittr (%$%) do wyciągnięcia wektora z ramki danych.

Na przykład:

iris2 %>% select(Species) %>% collect() %$% Species
Myślałem, że zasługuje na własną odpowiedź.
 12
Author: rrs,
Warning: date(): Invalid date.timezone value 'Europe/Kyiv', we selected the timezone 'UTC' for now. in /var/www/agent_stack/data/www/doraprojects.net/template/agent.layouts/content.php on line 54
2017-03-30 17:50:15