如何在R自定义函数中传入可选数据框列并检测其存在

在dplyr自定义函数里直接用is.null()检测可选列参数会报错，原因是未传入参数时R会尝试解析不存在的对象。下面提供两种可靠的解决方法：

方法1：捕获参数表达式判断是否传入 #

用enquo()捕获可选列的表达式，再通过quo_is_null()判断用户是否传入了该参数，避免直接解析不存在的对象：

library(dplyr)

my_calc_func <- function(df, x, y = NULL) {
  # 捕获y的表达式，不立即解析
  y_expr <- enquo(y)
  
  if (!quo_is_null(y_expr)) {
    # 传入y时执行乘法
    df %>% mutate(new_col = {{x}} * {{y}})
  } else {
    # 未传入y时直接使用x
    df %>% mutate(new_col = {{x}})
  }
}

# 测试场景1：不传入y参数
df_no_y <- tibble(x = c(1, 2, 3))
my_calc_func(df_no_y, x)

# 测试场景2：传入y参数
df_with_y <- tibble(x = c(1, 2, 3), y = c(4, 5, 6))
my_calc_func(df_with_y, x, y)

方法2：检查数据框中是否存在目标列 #

如果需求是不管是否显式传参，只要数据框包含指定列就执行乘法，可以用has_name()（dplyr内置函数）检查列是否存在：

my_calc_func2 <- function(df, x, y_col = NULL) {
  if (!is.null(y_col) && has_name(df, y_col)) {
    df %>% mutate(new_col = {{x}} * .data[[y_col]])
  } else {
    df %>% mutate(new_col = {{x}})
  }
}

# 测试场景1：数据框含y列，传入列名参数
my_calc_func2(df_with_y, x, "y")

# 测试场景2：数据框无y列，不传列名参数
my_calc_func2(df_no_y, x)

为什么之前的方法报错？ #

!is.null(y)或!is.null({{y}})会直接尝试解析y对象，当用户未传入该参数时，R找不到对应的对象，因此抛出object 'y' not found错误。而enquo()是捕获未计算的表达式，不会触发对象解析，因此能安全判断参数是否传入。

内容的提问来源于stack exchange，提问作者Jaken