如何在R函数中将输入变量名设为输出列V5的值？

直接通过substitute()捕获调用函数时传入的实际变量名，就能解决V5列固定显示"input"的问题，具体实现如下：

基础函数实现 #

result <- function(input) {
  # 捕获传入的数据集变量名
  source_label <- deparse(substitute(input))
  # 添加为新列V5
  input$V5 <- source_label
  return(input)
}

测试验证 #

# 创建测试数据集
input_1 <- data.frame(V1 = 1:3, V2 = 4:6)
input_2 <- data.frame(V1 = 7:9, V2 = 10:12)

# 调用函数处理
output_1 <- result(input_1)
output_2 <- result(input_2)

# 查看结果
output_1
#   V1 V2     V5
# 1  1  4 input_1
# 2  2  5 input_1
# 3  3  6 input_1

output_2
#   V1 V2     V5
# 1  7 10 input_2
# 2  8 11 input_2
# 3  9 12 input_2

批量处理多个数据集场景 #

如果需要一次性处理多个数据集，可以结合mget()和lapply()批量添加来源标签：

# 定义需要处理的数据集名称列表
dataset_list <- c("input_1", "input_2")

# 批量处理
processed_data <- lapply(dataset_list, function(name) {
  df <- get(name)
  df$V5 <- name
  df
})

# 给结果列表命名，方便后续调用
names(processed_data) <- dataset_list

关键说明 #

deparse(substitute(input))的作用是把函数调用时传入的实际对象名称转换成字符串，而不是函数定义里的参数名"input"。如果之前的代码没生效，大概率是因为在嵌套环境（比如循环内部）调用时没有正确捕获变量名，上述两种方法都能避免这个问题。

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