dplyr使用{{}}实现非标准求值时，如何动态调用带custom_前缀的函数

之前代码触发报错的核心原因来自两点：

• {{}}的注入能力仅在动态命名场景生效，也就是赋值左侧用"前缀_{{变量}}" :=生成动态列名的语法，是rlang为dplyr设计的专属动态命名规则
• 你尝试在函数调用位置使用"custom_{{func}}"({{colname}})时，R会直接将这段拼接后的字符串当作字面函数名查找，不会先对字符串做解析匹配对应的函数，因此会提示找不到名为"custom_{{func}}"的对象

实现逻辑 #

要满足用户仅传入mean/sum这类简短函数名、自动匹配对应custom_前缀自定义函数的需求，需要完成三步核心操作：

• 把用户传入的函数名转为字符串形式，拼接custom_前缀，从环境中检索对应的函数对象
• 沿用已验证可行的动态列名写法，让输出列自动生成custom_ + 列名的命名格式
• 借助rlang::exec()适配dplyr管道的非标准求值规则，完成函数的实际计算

完整代码 #

library(dplyr)
library(rlang)

# 预定义的两个带custom_前缀的自定义函数
custom_mean <- function(x) {
  mean(x, na.rm = TRUE)
}

custom_sum <- function(x) {
  sum(x, na.rm = TRUE)
}

# 合并后的通用处理函数
apply_on_mtcars <- function(colname, func) {
  # 解析用户传入的函数名，拼接前缀后取出对应函数
  func_sym <- as_string(ensym(func))
  target_func <- get(paste0("custom_", func_sym), envir = globalenv())

  mtcars %>%
    summarise(
      # 原有动态列名逻辑保持不变
      "custom_{{colname}}" := exec(target_func, {{colname}})
    )
}

调用示例 #

完全符合低门槛的调用要求，用户无需感知custom_前缀的存在：

# 计算drat列的均值
apply_on_mtcars(drat, mean)
#   custom_drat
# 1    3.596563

# 计算drat列的总和
apply_on_mtcars(drat, sum)
#   custom_drat
# 1     115.09

• 后续如果新增其他同规则的自定义函数（如custom_median），不需要修改通用函数的代码，用户直接传入median参数即可完成对应计算
• ensym()可以同时兼容用户传入裸函数名（如mean）、字符串形式函数名（如"mean"）两种写法，鲁棒性更强
• 不要尝试直接把拼接的字符串放在函数调用位置，R的语法解析流程不会对字符串形式的函数名做动态查找

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