r - 测试使用 enquo() 作为 NULL 参数的函数

Question

我有一个创建数据框的函数，但在此过程中更改了名称。我正在尝试使用 dplyr quosures 处理空列名。我的测试套件如下所示:

dataframe <- data_frame(
  a = 1:5,
  b = 6:10
)

my_fun <- function(df, col_name, new_var_name = NULL) {
  target <- enquo(col_name)

  c <- df %>% pull(!!target) * 3 # here may be more complex calculations

  # handling NULL name
  if (is.null(new_var_name)) {
    new_name <- quo(default_name)
  } else{
    new_name <- enquo(new_name)
  }

  data_frame(
    abc = df %>% pull(!!target),
    !!quo_name(new_name) := c
  )
}

如果我这样调用我的函数:

my_fun(dataframe, a)

我按预期获得默认名称:

# A tibble: 5 x 2
    abc default_name
  <int>        <dbl>
1     1            3
2     2            6
3     3            9
4     4           12
5     5           15

如果我尝试传递名称，则会出现错误:

my_fun(dataframe, a, NEW_NAME)
Error in my_fun(dataframe, a, NEW_NAME) : object 'NEW_NAME' not found

我哪里错了？

Answer 1

这个问题实际上与 quo 无关。和 enquo返回不同的东西，这真的是关于在你真正想要之前评估对象。如果您要使用 browser()要逐步执行您的函数，您会看到错误发生在 if (is.null(new_var_name))陈述。

当您这样做时is.null(new_var_name) ，您正在评估作为 new_var_name 传递的变量所以现在为时已晚 enquo它。那是因为 is.null需要查看变量的值而不仅仅是变量名称本身。

一个不计算传递给函数的参数但检查它是否存在的函数 missing() .

my_fun <- function(df, col_name, new_var_name=NULL) {
  target <- enquo(col_name)

  c <- df %>% pull(!!target) * 3 # here may be more complex calculations

  # handling NULL name
  if (missing(new_var_name)) {
    new_name <- "default_name"
  } else{
    new_name <- quo_name(enquo(new_var_name))
  }

  data_frame(
    abc = df %>% pull(!!target),
    !!new_name := c
  )
}

然后你可以运行这两个

my_fun(dataframe, a)
my_fun(dataframe, a, NEW_NAME)