Trích xuất một cột dplyr tbl như một vector

Có cách nào ngắn gọn hơn để lấy một cột của một dplyr tbl làm vector, từ một tbl với back-end cơ sở dữ liệu (tức là khung / bảng dữ liệu không thể được đặt trực tiếp vào tập hợp con)?

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

Điều đó đã quá dễ dàng, vì vậy

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

Nhưng có vẻ hơi vụng về.

Với dplyr 0.7.0, bạn có thể sử dụng pullđể lấy vectơ từ a 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"

Theo nhận xét từ @nacnudus, có vẻ như một pullchức năng đã được triển khai trong dplyr 0.6:

iris2 %>% pull(Species)

Đối với các phiên bản cũ hơn của dplyr, đây là một chức năng gọn gàng để giúp kéo ra một cột đẹp hơn một chút (dễ gõ hơn và dễ đọc hơn):

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

Điều này cho phép bạn thực hiện một trong những điều sau:

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

Kết quả là ...

 [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

Và nó cũng hoạt động tốt với các khung dữ liệu:

> 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

Một cách hay để làm điều này trong v0.2 của dplyr:

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

Hoặc nếu bạn thích:

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

Hoặc nếu bàn của bạn không quá lớn, chỉ cần ...

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

Bạn cũng có thể sử dụng unlistnhững gì tôi thấy dễ đọc hơn vì bạn không cần lặp lại tên của cột hoặc chỉ định chỉ mục.

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

Tôi sẽ sử dụng extract2chức năng tiện lợi từ magrittr:

library(magrittr)
library(dplyr)

iris2 %>%
  select(Species) %>%
  extract2(1)  

Tôi có thể viết:

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

Vì dplyr được thiết kế để làm việc với các tbl dữ liệu, không có cách nào tốt hơn để có được một cột dữ liệu.

@ Luke1018 đề xuất giải pháp này trong một trong những ý kiến:

Bạn cũng có thể sử dụng magrittrtoán tử giải thích ( %$%) để kéo một vectơ từ khung dữ liệu.

Ví dụ:

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

Tôi nghĩ rằng nó xứng đáng câu trả lời của riêng mình.

Nếu bạn quen sử dụng dấu ngoặc vuông để lập chỉ mục, một tùy chọn khác là chỉ bao bọc cách tiếp cận lập chỉ mục thông thường trong lệnh gọi tới deframe () , ví dụ:

library(tidyverse)

iris2 <- as_tibble(iris)

# using column name
deframe(iris2[, 'Sepal.Length'])

# [1] 5.1 4.9 4.7 4.6 5.0 5.4

# using column number
deframe(iris2[, 1])

# [1] 5.1 4.9 4.7 4.6 5.0 5.4

Điều đó và pull () đều là những cách khá hay để có được một cột gọn gàng.