データベースバックエンドを持つtblから、dplyr tblの1列をベクトルとして取得するより簡潔な方法はありますか(つまり、データフレーム/テーブルを直接サブセット化することはできません)?
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.
しかし、それは少し不器用なようです。
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"
@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
dplyr
のv0.2でこれを行う良い方法:
iris2 %>% select(Species) %>% collect %>% .[[5]]
または、必要に応じて:
iris2 %>% select(Species) %>% collect %>% .[["Species"]]
または、テーブルが大きすぎない場合は、単に...
iris2 %>% collect %>% .[["Species"]]
unlist
を使用することもできます。これは、列の名前を繰り返したり、インデックスを指定したりする必要がないため、読みやすくなっています。
iris2 %>% select(Species) %>% unlist(use.names = FALSE)
magrittr
のextract2
便利な関数を使用します。
library(magrittr)
library(dplyr)
iris2 %>%
select(Species) %>%
extract2(1)
私はおそらく書くでしょう:
collect(select(iris2, Species))[[1]]
Dplyrはデータのtblを操作するために設計されているため、データの単一の列を取得するより良い方法はありません。
@ Luke1018は、コメントの1つでこのソリューションを提案しました。
magrittr
エクスポジション演算子(%$%
)を使用して、データフレームからベクターをプルすることもできます。
例えば:
iris2 %>% select(Species) %>% collect() %$% Species
独自の答えに値すると思いました。