如何在函数if语句的dplyr管道中传递值并处理拼接列名？

首先准备数据集：

library(dplyr)
library(ggplot2) # msleep 数据集来自 ggplot2

d <- msleep %>% 
      mutate(comfort_color_desk = sample(c(0,1), replace=TRUE, size=nrow(msleep)),
             comfort_color_chair = sample(c(0,1), replace=TRUE, size=nrow(msleep))) 

需求是编写自定义函数，传入列名后缀（如desk、chair），拼接成完整列名后在dplyr链式调用的filter中使用，同时通过指示器.compare触发条件过滤。但以下函数无法正常运行：

test_2_fcn <- function(.x, .y, .z, .compare = NULL) {
  .x %>% 
    {if(isTRUE(.compare)) filter(., paste0("comfort_color_", !!ensym(.y)) == 1 & paste0("comfort_color", !!ensym(.z)) == 1) else .} 
}

d %>% test_2_fcn(., .y = "desk", .z = "chair", .compare = TRUE)

1. 字符串未被解析为列引用：paste0生成的是普通字符串，直接和1比较时，是判断字符串与数字是否相等，而非引用数据框中对应列的数值。
2. ensym使用冗余：!!ensym(.y)会将传入的字符串转为符号，但paste0会强制把符号转回字符串，这一步完全多余，反而混淆了列引用逻辑。
3. 列名拼接错误：原代码中paste0("comfort_color", !!ensym(.z))少了下划线，会生成错误的列名comfort_colorchair，而非正确的comfort_color_chair。

方法1：用sym()转换列名为符号并注入 #

先拼接完整列名字符串，再用sym()转为dplyr能识别的符号，最后用!!注入到过滤条件中：

test_2_fcn <- function(.x, .y, .z, .compare = NULL) {
  # 拼接完整列名并转为符号
  col_y <- sym(paste0("comfort_color_", .y))
  col_z <- sym(paste0("comfort_color_", .z))
  
  .x %>% 
    {if(isTRUE(.compare)) filter(., !!col_y == 1 & !!col_z == 1) else .} 
}

# 测试运行
d %>% test_2_fcn(.y = "desk", .z = "chair", .compare = TRUE)

方法2：使用.data代词（更简洁） #

dplyr的.data代词可以直接通过字符串引用列，无需符号转换，代码更直观：

test_2_fcn <- function(.x, .y, .z, .compare = NULL) {
  col_y <- paste0("comfort_color_", .y)
  col_z <- paste0("comfort_color_", .z)
  
  .x %>% 
    {if(isTRUE(.compare)) filter(., .data[[col_y]] == 1 & .data[[col_z]] == 1) else .} 
}

# 测试运行
d %>% test_2_fcn(.y = "desk", .z = "chair", .compare = TRUE)

方法3：用across简化多列条件（可选） #

如果需要批量处理类似命名的列，也可以用across结合模式匹配来实现：

test_2_fcn <- function(.x, .y, .z, .compare = NULL) {
  col_pattern <- paste0("comfort_color_(", .y, "|", .z, ")")
  
  .x %>% 
    {if(isTRUE(.compare)) filter(., across(all_of(col_pattern), ~ .x == 1)) else .} 
}

# 测试运行
d %>% test_2_fcn(.y = "desk", .z = "chair", .compare = TRUE)

内容的提问来源于stack exchange，提问作者C.Robin