R语言如何向接收函数为参数的自定义函数正确传递参数

错误原因 #

代码的核心问题出在函数默认参数的求值逻辑和lm/lm_robust的参数求值环境冲突：

• 你给weights设置的默认值是wgt、clusters默认值是cl，但R函数的默认参数是在函数被调用的环境（而非函数内部环境）求值的，当你调用reg时没有显式传入这两个参数、同时全局环境中不存在wgt/cl对象时，就会抛出找不到对象的错误。
• 另外lm系列函数本身会在公式对应的数据环境里求值权重、聚类变量，直接传未求值的符号会进一步触发环境匹配问题。

修复方案 #

推荐两种修复思路，按需选择：

思路1：将可选参数默认值设为NULL，仅在用户显式传入时才调用对应参数 #

这种方案兼容性最高，不需要依赖全局对象：

library(estimatr)

reg <- function(fm, data = df, f = lm, weights = NULL, 
                clusters = NULL, se = 'stata') {
  # 构造基础参数列表
  args <- list(
    formula = as.formula(fm),
    data = data
  )
  # 仅当用户传入对应参数时才添加到调用列表
  if (!is.null(weights)) args$weights <- substitute(weights)
  if (!is.null(clusters)) args$clusters <- substitute(clusters)
  if (identical(f, lm_robust)) args$se <- se
  
  # 调用对应函数
  m <- do.call(f, args)
  
  # 返回结果
  return(m)
}

调用示例

# 不需要权重、聚类的普通lm调用，不会报错
reg(y ~ x1 + x2 + x3, data = your_data, f = lm)

# 带权重的lm_robust调用
reg(y ~ x1 + x2 + x3, data = your_data, f = lm_robust, weights = wgt_col, clusters = cl_col)

思路2：保留默认参数名，调整求值逻辑 #

如果你确定默认的权重列名就是wgt、聚类列名就是cl，可以直接把默认值设为字符串，从数据框里直接取对应列：

library(estimatr)

reg <- function(fm, data = df, f = lm, weights = "wgt", 
                clusters = "cl", se = 'stata') {
  args <- list(
    formula = as.formula(fm),
    data = data
  )
  # 检查数据里是否存在默认列，存在才添加参数
  if (weights %in% colnames(data)) args$weights <- data[[weights]]
  if (clusters %in% colnames(data) && identical(f, lm_robust)) args$clusters <- data[[clusters]]
  if (identical(f, lm_robust)) args$se <- se
  
  m <- do.call(f, args)
  return(m)
}

核心修复逻辑说明 #

• 用do.call构造函数调用，避免硬编码分支的参数匹配问题
• 用substitute捕获用户传入的变量名，适配lm系列函数的参数求值规则
• 把可选参数默认值设为NULL/明确的字符串，避免未定义对象的求值错误

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