如何实现支持灵活新增列的myfun函数？解决glue调用报错问题

问题背景 #

需要编写一个myfun函数，支持基于传入的data.frame灵活创建新列，但使用dplyr::mutate结合glue处理字符列时，出现外部变量（如session）找不到的报错，核心需求是灵活新增列而非修改原数据框。

报错重现代码 #

library(dplyr)
library(glue)

# 存在问题的函数版本
myfun <- function(df, new_col_expr) {
  df %>%
    mutate(!!sym("new_col") := glue({new_col_expr}))
}

# 测试数据与外部变量
test_df <- data.frame(name = c("Alice", "Bob"), age = c(25, 30))
session <- "2024"

# 调用时触发报错
myfun(test_df, "Name: {name}, Age: {age}, Session: {session}")

错误信息 #

Error in glue_data(.x, .call = .call, .envir = .envir, ...) : 
  object 'session' not found

问题根源 #

glue默认使用当前函数的内部环境，而session定义在函数外部的调用环境中，直接使用无法捕获该变量；同时原函数的非标准求值处理方式未正确传递环境上下文。

解决方案 #

通过传递调用环境给glue，结合rlang的非标准求值工具，实现既能捕获外部变量，又能灵活自定义新列名和生成逻辑的函数：

最终实现代码 #

library(dplyr)
library(glue)
library(rlang)

myfun <- function(df, new_col_name, new_col_expr) {
  # 获取调用环境，确保能访问外部定义的变量
  caller_env <- caller_env()
  
  df %>%
    mutate(
      !!sym(new_col_name) := glue({{new_col_expr}}, .envir = caller_env)
    )
}

输入输出示例 #

输入

test_df <- data.frame(name = c("Alice", "Bob"), age = c(25, 30))
session <- "2024"

# 调用函数：指定新列名，传入包含数据框列和外部变量的glue表达式
result_df <- myfun(test_df, "info", "Name: {name}, Age: {age}, Session: {session}")

输出

name age                          info
1 Alice  25 Name: Alice, Age: 25, Session: 2024
2   Bob  30   Name: Bob, Age: 30, Session: 2024

扩展版本：支持直接传入dplyr表达式 #

如果希望用户无需将表达式包裹成字符串，可使用以下版本，直接支持dplyr风格的表达式：

myfun2 <- function(df, new_col_name, new_col_expr) {
  df %>%
    mutate(!!sym(new_col_name) := {{new_col_expr}})
}

# 使用示例
result_df2 <- myfun2(test_df, "info_str", 
                     stringr::str_c("Name: ", name, ", Age: ", age, ", Session: ", session))

这个版本同样能正确捕获外部的session变量，且更符合dplyr的使用习惯。

内容的提问来源于stack exchange，提问作者Fredrik Nylén