R语言函数编写：如何同时引用列名与对应字符串前缀进行移除？

要解决这个问题，核心是利用tidyverse的**非标准求值（NSE）**语法，让函数既能识别列名，又能自动生成对应前缀。以下是具体实现方案：

1. 依赖包准备 #

确保先加载所需的包：

library(dplyr)
library(stringr)

2. 单列处理函数 #

先写一个针对单列的可复用函数，它能自动获取列名并生成对应的前缀：

remove_col_prefix <- function(data, col) {
  # 将传入的列名参数转换为字符串，用于构造前缀
  col_str <- as_label(enquo(col))
  prefix <- str_c(col_str, "_")
  
  # 修改目标列，移除前缀
  data %>%
    mutate({{ col }} := str_remove({{ col }}, prefix))
}

使用示例 #

先构造测试数据：

test_df <- tibble(
  y = c("y_apple", "y_banana", "y_cherry"),
  z = c("z_dog", "z_cat", "z_bird")
)

调用函数处理单列，可链式调用处理多列：

cleaned_df <- test_df %>%
  remove_col_prefix(y) %>%
  remove_col_prefix(z)

# 输出结果
cleaned_df

3. 多列批量处理函数 #

如果需要一次处理多列，可优化函数支持批量操作：

remove_multiple_prefixes <- function(data, ...) {
  data %>%
    mutate(across(c(...), ~ str_remove(., str_c(cur_column(), "_"))))
}

使用示例 #

cleaned_df2 <- test_df %>% remove_multiple_prefixes(y, z)
cleaned_df2

原理说明 #

• enquo(col) + {{ col }}：捕获传入的列名表达式，让函数能识别数据框中的列（而非普通变量）。
• as_label(enquo(col))：将捕获的列名表达式转换为字符串，用于生成对应前缀（比如列名y转为字符串"y"，再拼接成"y_"）。
• cur_column()：在across中使用时，自动获取当前处理的列名，实现批量生成对应前缀。

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