如何实现修改R语言泛型函数默认na.rm参数的函数工厂？

首先纠正原代码中的明显逻辑错误：判断函数是否包含na.rm参数的条件写反了，正确的判断应该是如果函数没有na.rm参数才抛出错误，修正后的代码段如下：

if(!"na.rm" %in% names(formals(f))){
  stop("fun `f` does not have `na.rm` as an argument")
}

针对泛型函数（如mean()）的适配问题，无需创建新的泛型类和方法，以下是两种更简洁的实现思路：

思路1：创建包装函数（通用且安全） #

这种方式不修改原函数或方法的参数默认值，而是生成一个包装函数，调用原函数时自动传入na.rm = TRUE作为默认值，同时保留用户手动覆盖该参数的能力。该方法对普通函数和泛型函数均适用。

实现代码：

fun0 <- function(f, env = globalenv()){
  fun <- match.fun(f)
  # 检查是否存在na.rm参数
  if(!"na.rm" %in% names(formals(fun))){
    stop("fun `f` does not have `na.rm` as an argument")
  }
  
  # 获取原函数名称
  fname <- if(is.function(f)) {
    as.character(substitute(f))
  } else {
    f
  }
  fname0 <- paste0(fname, "0")
  
  # 创建包装函数
  wrapper <- function(...) {
    args <- list(...)
    # 用户未传入na.rm时，自动添加默认值TRUE
    if(!"na.rm" %in% names(args)){
      args$na.rm <- TRUE
    }
    do.call(fun, args)
  }
  
  # 复制原函数的元信息（可选，让新函数更贴近原函数的表现）
  formals(wrapper) <- formals(fun)
  formals(wrapper)$na.rm <- TRUE
  attr(wrapper, "srcref") <- attr(fun, "srcref")
  class(wrapper) <- class(fun)
  
  # 将新函数赋值到目标环境
  assign(fname0, wrapper, envir = env)
}

测试示例：

fun0(mean)
mean0(c(1, 2, NA))          # 输出1.5，符合预期
mean0(c(1, 2, NA), na.rm = FALSE) # 输出NA，支持手动覆盖参数

思路2：修改泛型函数的默认方法参数（泛型特定方案） #

如果仅针对泛型函数优化，可以直接修改其默认方法的na.rm参数默认值，新生成的泛型函数复用原泛型的方法调度逻辑。这种方式更贴近原代码的思路，但仅对泛型函数有效。

实现代码：

fun0 <- function(f, env = globalenv()){
  fun <- match.fun(f)
  # 检查是否存在na.rm参数
  if(!"na.rm" %in% names(formals(fun))){
    stop("fun `f` does not have `na.rm` as an argument")
  }
  
  fname <- if(is.function(f)) {
    as.character(substitute(f))
  } else {
    f
  }
  fname0 <- paste0(fname, "0")
  
  # 处理泛型函数：修改默认方法的na.rm默认值
  if(isGeneric(fname)){
    default_method <- get(paste0(fname, ".default"), envir = environment(fun))
    formals(default_method)$na.rm <- TRUE
    # 创建新泛型并注册修改后的默认方法
    new_generic <- fun
    assign(fname0, new_generic, envir = env)
    setMethod(fname0, signature = getGeneric(fname)@signature, definition = default_method)
  } else {
    # 非泛型函数沿用原逻辑
    formals(fun)$na.rm <- TRUE
    assign(fname0, fun, envir = env)
  }
}

测试示例：

fun0(mean)
mean0(c(1, 2, NA))          # 输出1.5
mean0(c(1, 2, NA), na.rm = FALSE) # 输出NA

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