将 `c(x, y)` 作为函数参数转换为 R 中的 `c("x", "y")` (rlang/tidyevaluation)

Question

我想让我的函数的用户以 tidyeval 数据屏蔽方式输入变量，即 fun(x, c(y, z)) 而不是 fun("x", c("y", "z"))，但是当他们输入像 c(y, z) 这样的参数时，我希望能够将其用作一个字符向量，例如c("y", "z")。

对于使用列名，我确实有一个严重过度的解决方法。也就是说，我在dpylr::select() 调用中使用{{，然后提取列名：

fun_overkill <- function(data, cols) {
  data_subset <- dplyr::select(data, {{cols}})
  
  args_as_character_vector <- colnames(data_subset)
  
  args_as_character_vector
}

fun_overkill(mtcars, c(mpg, drat))

显然，这不是最好的方法，并且它不能推广到我的向量参数不代表列名的情况。

我怀疑{rlang} 可能有答案，但有很多选择，我不懂术语。

在创建模型公式时可能会有用，因为您可以在从字符向量创建的字符串上使用 as.formula：

rhs <- paste(fun_overkill(mtcars, c(mpg, drat)), collapse = " + ")
lhs <- "~ "
as.formula(paste0(lhs, rhs))

Answer 1

一种方法是使用enexpr() 捕获表达式的cols 变量（enquo() 也可以），然后使用call_args() 从表达式中获取参数。这些现在是符号列表，因此您需要循环它们以使它们成为带有as_string() 的字符。

library(rlang)

fun <- function(data, cols) {
  vars <- call_args(enexpr(cols))
  vapply(vars, as_string, character(1), USE.NAMES = FALSE)
}

(fun(mtcars, c(mpg, drat)))
#> [1] "mpg"  "drat"


rhs <- paste(fun(mtcars, c(mpg, drat)), collapse = " + ")
lhs <- "~ "
as.formula(paste0(lhs, rhs))
#> ~mpg + drat

^{由reprex package (v2.0.1) 于 2021 年 9 月 3 日创建}

公平地说，这里根本不需要 data 参数。它就在那里，因为它在您的示例中就在那里。