如何在R函数中判断传入的函数名参数并执行差异化处理？

这问题我之前也处理过，核心就是要先捕获传入函数的名称，再根据不同名称执行差异化逻辑。下面是完整的实现代码和说明：

完整函数代码 #

fun1 <- function(x, fun2) {
  # 捕获传入的函数名称并转为字符串
  fun_name <- deparse(substitute(fun2))
  
  # 根据函数名执行对应逻辑
  result <- switch(fun_name,
                   "mean" = {
                     # 针对mean：返回向量（支持向量、矩阵、数据框输入）
                     if (is.matrix(x) || is.data.frame(x)) {
                       apply(x, 2, mean, na.rm = TRUE)
                     } else {
                       as.vector(mean(x, na.rm = TRUE))
                     }
                   },
                   "summary" = {
                     # 针对summary：返回列表形式的结果
                     as.list(summary(x))
                   },
                   # 默认情况：直接执行传入的函数并返回原结果
                   {
                     fun2(x)
                   })
  
  return(result)
}

关键逻辑解释 #

• 获取函数名称：用deparse(substitute(fun2))是因为当你把函数（比如mean）作为参数传入时，实际传入的是函数对象，substitute(fun2)会捕获传入的函数名表达式，再通过deparse转成字符串，这样就能判断是哪个函数了。
• 分支判断：用switch比多个if-else更简洁，适合这种基于固定字符串的分支场景。
• 差异化处理：
  • 当fun2是mean时：如果输入是矩阵或数据框，就按列计算均值返回向量；如果是普通向量，就把均值转成向量形式返回。
  • 当fun2是summary时：把默认返回的summary结果转成列表格式。
  • 其他函数：直接执行传入的函数，返回原结果。

测试示例 #

测试mean场景 #

# 普通向量输入
x1 <- c(1, 2, 3, 4, 5)
fun1(x1, mean)
# 输出：[1] 3 （向量形式）

# 数据框输入
x2 <- data.frame(a = 1:5, b = 6:10)
fun1(x2, mean)
# 输出：a b 
#      3 8 （向量形式）

测试summary场景 #

fun1(x1, summary)
# 输出：
# $Min.
# [1] 1
# 
# $`1st Qu.`
# [1] 2
# 
# $Median
# [1] 3
# 
# $Mean
# [1] 3
# 
# $`3rd Qu.`
# [1] 4
# 
# $Max.
# [1] 5 （列表形式）

测试其他函数（默认逻辑） #

fun1(x1, sum)
# 输出：[1] 15

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