Напишите свои собственные функции tidyselect

Question

Я написал пакет R, который использует селекторы {tidyselect} (например, contains(), starts_with(), и т.д.). Я хотел бы добавить еще несколько вспомогательных функций select в пакет для выбора переменных на основе некоторого атрибута. Например, выберите все числовые переменные или, возможно, все логические переменные.

Я рассмотрел базовый код {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

Любые советы или указания к какой-либо документации будут с благодарностью! Спасибо!

Answer 1

Когда вы звоните 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

^{Создано в 2019-06-13 пакетом представлением (v0.2.1)}