3

我编写了一个使用 {tidyselect} 选择器(例如,等)的 Rcontains()starts_with()。我想在包中添加更多选择帮助函数,以根据某些属性选择变量。例如,选择所有数值变量或可能所有逻辑变量。

我已经查看了 {tidyselect} 基本代码。但我无法推测变量的注册是如何工作的,因此无法将其扩展到通过属性选择变量。

我做了一些搜索,看起来 {recipes} 包已经成功实现了我正在寻找的其他帮助程序(例如all_numeric()),但我自己正在努力编写扩展函数。https://github.com/tidymodels/recipes/blob/master/R/selections.R

我相信,归根结底是我不明白当变量注册到tidyselect::scoped_vars()函数时会发生什么。如果我tidyselect::scoped_vars(vars = names(mtcars))在干净的环境中运行,我看不到任何更改。但是我可以在注册变量后在全局环境中使用 {tidyselect} 助手。

names(mtcars)
#>  [1] "mpg"  "cyl"  "disp" "hp"   "drat" "wt"   "qsec" "vs"   "am"   "gear"
#> [11] "carb"
tidyselect::scoped_vars(vars = names(mtcars))

# returns position of column 'mpg'
tidyselect::starts_with("mp")
#> 1

任何提示或对某些文档的指导将不胜感激!谢谢!

4

1 回答 1

2

当您调用scoped_vars()时,给定的变量名称在当前函数调用期间保存在内部环境中:

(function() {
  print(tidyselect:::vars_env$selected)
  tidyselect::scoped_vars(names(mtcars))
  print(tidyselect:::vars_env$selected)
})()
#> NULL
#>  [1] "mpg"  "cyl"  "disp" "hp"   "drat" "wt"   "qsec" "vs"   "am"   "gear"
#> [11] "carb"

print(tidyselect:::vars_env$selected)
#> NULL

据我所知,这是 {tidyselect} 保留的有关变量的唯一信息;所以如果要根据属性进行选择,则必须自己维护属性信息。这也是 {recipes} 在cur_info_env环境中所做的。

粗略的实现可能如下所示:

type_env <- rlang::new_environment()

select_with_attributes <- function(.data, ...) {
  type_env$types <- purrr::map(.data, class)
  dplyr::select(.data, ...)
}

all_numeric <- function() {
  which(purrr::map_lgl(type_env$types, ~ any(.x %in% "numeric")))
}

head(select_with_attributes(iris, all_numeric()))
#>   Sepal.Length Sepal.Width Petal.Length Petal.Width
#> 1          5.1         3.5          1.4         0.2
#> 2          4.9         3.0          1.4         0.2
#> 3          4.7         3.2          1.3         0.2
#> 4          4.6         3.1          1.5         0.2
#> 5          5.0         3.6          1.4         0.2
#> 6          5.4         3.9          1.7         0.4

reprex 包(v0.2.1)于 2019 年 6 月 13 日创建

于 2019-06-13T15:28:56.137 回答