Извлечение столбца dplyr tbl в качестве вектора

Question

Извлечение столбца dplyr tbl в качестве вектора

есть ли более краткий способ получить один столбец dplyr tbl в качестве вектора из tbl с базой данных back-end (т. е. фрейм данных/таблица не может быть подмножеством напрямую)?

require(dplyr)

db <- src_sqlite(tempfile(), create = TRUE)

iris2 <- copy_to(db, iris)

iris2$Species

# NULL

Это было бы слишком просто, поэтому

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

# [1] "setosa"     "setosa"     "setosa"     "setosa"  etc.

но это кажется немного неуклюжим.

666 6

r lazy-evaluation collect

6 ответов:

Comments

Ничего не найдено.

Lorenz Walthert · Accepted Answer · 2017-06-14 13:49:28

С dplyr 0.7.0, вы можете использовать pull чтобы получить вектор от 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"

Tommy O'Dell · Accepted Answer · 2017-12-05 19:48:27

согласно комментарию от @nacnudus, это выглядит как pull функция была реализована в dplyr 0.6:
iris2 %>% pull(Species)
для более старых версий dplyr, вот аккуратная функция, чтобы сделать вытаскивание столбца немного приятнее (легче печатать и легче читать):
pull <- function(x,y) {x[,if(is.name(substitute(y))) deparse(substitute(y)) else y, drop = FALSE][[1]]}
это позволяет вам сделать любой из этих:
iris2 %>% pull('Species')
iris2 %>% pull(Species)
iris2 %>% pull(5)
результате...
 [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
и он также отлично работает с фреймами данных:
> 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
A хороший способ сделать это в v0. 2 из dplyr:
iris2 %>% select(Species) %>% collect %>% .[[5]]
или, если вы предпочитаете:
iris2 %>% select(Species) %>% collect %>% .[["Species"]]
или если ваш стол не слишком большой, просто...
iris2 %>% collect %>% .[["Species"]]

StanislawSwierc · Accepted Answer · 2015-06-27 13:02:05

вы также можете использовать unlist который мне легче читать, потому что вам не нужно повторять имя столбца или указывать индекс.
iris2 %>% select(Species) %>% unlist(use.names = FALSE)

Hugh · Accepted Answer · 2014-11-26 11:20:05

Я бы использовал extract2 функции комфорта от magrittr:
library(magrittr)
library(dplyr)

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

hadley · Accepted Answer · 2014-02-07 15:33:19

Я бы, наверное, написать:
collect(select(iris2, Species))[[1]]
поскольку dplyr предназначен для работы с tbls данных, нет лучшего способа получить один столбец данных.

rrs · Accepted Answer · 2017-03-30 20:50:15

@Luke1018 предложил это решение в одном из комментариев:

вы также можете использовать magrittr экспозиция оператор (%$%) для извлечения вектора из фрейма данных.

например:
iris2 %>% select(Species) %>% collect() %$% Species
Я думал, что он заслужил свой собственный ответ.