如何创建同时支持引用和未引用参数的tidyverse函数？

Question

我知道如何创建函数来支持名为“变量”的参数的准引用{使用dplyr::e（变量）作为未加引号的函数参数}或需要您引用参数的函数{使用rlang::sym（变量）}。有没有一种简单的方法可以使函数既支持加引号又支持未加引号的参数？

例如，dplyr::select（）允许select（mtcar， mpg）和select（mtcar，"mpg"）。构建一个可以做这两者的函数的最佳实践是什么？一个考虑因素是对数据屏蔽的影响，我不确定在构建更复杂的函数时是否需要考虑这一点。

我一直在github页面上寻找基本的dplyr函数，但是像select这样的简单函数依赖于一个全新的包（tidyselect），所以有很多事情要做。我在Tidy评估书中也没有看到明确的解释。下面是一个同时支持引用和未引用参数的hack函数，但这不是一个可靠的解决方案。我相信有一个更简单的方法。

library(dplyr)

data(mtcars)

test_func <- function(variable) {
  if(nrow(count(mtcars, {{variable}})) == 1) {
    variable <- rlang::sym(variable)
  }
  count(mtcars, {{variable}})
}

all_equal(
  test_func(cyl),
  test_func("cyl")
)

Answer 1

如果需要对引用/未引用都起作用，请使用< code>ensym

test_func <- function(variable) {  

    dplyr::count(mtcars, !!rlang::ensym(variable))    

  }

-测试

test_func(cyl)
#  cyl  n
#1   4 11
#2   6  7
#3   8 14
test_func('cyl')
#  cyl  n
#1   4 11
#2   6  7
#3   8 14

注意：最好将数据也作为函数的参数