如何编写R函数按前缀批量选取数据框中的对应列？

你的函数核心问题是没有将参数count_char的值与后缀.x/.y拼接成实际的列名，直接写count_char.x会被R当作字面量列名（即找名为count_char.x的列），而非你传入的前缀拼接后的结果。以下是两种可行的解决方法：

方法1：支持传入裸名参数（如count_A） #

使用tidy eval语法将裸名参数转换为字符串，再拼接列名：

library(dplyr)

pull_columns2 <- function(df, count_char) {
  # 将传入的裸名参数转为字符串前缀
  prefix <- rlang::as_name(rlang::enquo(count_char))
  # 生成目标列名
  target_cols <- c(paste0(prefix, ".x"), paste0(prefix, ".y"))
  # 选择对应列
  df %>% select(all_of(target_cols))
}

调用示例：

# 直接传裸名count_A即可
pull_columns2(df, count_A)

方法2：支持传入字符串参数（如"count_A"） #

如果习惯传字符串形式的前缀，函数可以更简洁：

library(dplyr)

pull_columns2 <- function(df, count_char) {
  target_cols <- c(paste0(count_char, ".x"), paste0(count_char, ".y"))
  df %>% select(all_of(target_cols))
}

调用示例：

# 传入字符串形式的前缀
pull_columns2(df, "count_A")

关键说明 #

• paste0()用于将前缀与后缀.x/.y拼接成完整列名
• all_of()用于告诉dplyr：传入的是实际的列名字符向量，而非变量名

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