如何用rlang双大括号编写支持裸列名的data.table函数？

需要编写支持**裸列名（如Sepal.Length）和字符串列名（如"Sepal.Length"）**两种传参方式的函数，同时使用data.table的na.omit保证处理效率。目前在tibble环境下两种传参都有效，但切换到data.table环境后，裸列名传参会失败，以下是具体解决方法：

一、tibble环境下的可行实现 #

在tibble中，dplyr::na.omit可以直接通过{{}}语法兼容两种列名传参方式：

library(data.table)
library(dplyr)

data("iris")
iris[1,1:2] = NA
iris[3,3:4] = NA
dt = as_tibble(iris)

test_tibble <- function(dt, col1) {
  dt = dt %>% 
    na.omit(cols = c({{col1}}))
  return(dt)
}

# 两种传参方式均有效
test_tibble(dt, Sepal.Length)
test_tibble(dt, "Sepal.Length")

二、data.table环境下的问题 #

直接将数据转为data.table后，裸列名传参会报错，仅字符串列名能正常运行：

test_dt <- function(dt, col1) {
  dt = data.table(iris)
  dt = dt %>% 
    na.omit(cols = c({{col1}}))
  return(dt)
}

# 字符串传参正常
test_dt(dt, "Sepal.Length")
# 裸列名传参失败
test_dt(dt, Sepal.Length)

三、解决方案：统一列名为字符串格式 #

核心思路是用rlang工具将裸列名转换为字符串，同时兼容字符串传参的情况，适配data.table的参数要求：

基础版：单列传参修正 #

test_dt_fixed <- function(dt, col1) {
  dt = data.table(dt)
  # 自动识别裸列名/字符串，转为统一的字符串格式
  col1_str <- rlang::as_string(rlang::ensym(col1))
  dt = dt %>% 
    na.omit(cols = c("Species", col1_str))
  return(dt)
}

# 两种传参方式均正常运行
test_dt_fixed(dt, Sepal.Length)
test_dt_fixed(dt, "Sepal.Length")

进阶版：支持多列传参（适配复杂场景） #

针对需要传入多列的场景，用tidyselect::vars_select()处理更灵活，支持单列、多列（如c(CRT1, CRT2)）的裸名/字符串传参：

library(tidyselect)

# 修正用户提供的复杂函数示例
f <- function(data, type, arg1, arg2){
  data = data.table(data)
  
  # 提取传入的列名，兼容裸名、字符串、多列组合
  arg1_cols <- tidyselect::vars_select(names(data), {{arg1}})
  arg2_cols <- tidyselect::vars_select(names(data), {{arg2}})
  
  if (type != "a") {
    keep_cols <- c("Y", "G", "T", "D", arg1_cols, arg2_cols)
    data <- data %>% na.omit(cols = keep_cols)
  } else {
    keep_cols <- c("Y", "T", "D", "D0", arg1_cols)
    data <- data %>% na.omit(cols = keep_cols)
  }
  data
}

# 测试：多列裸名传参
f(tb2_2, type= "a", arg1 = c(CRT1, CRT2), arg2 = TREAT1)
# 测试：字符串传参
f(tb2_2, type= "a", arg1 = c("CRT1", "CRT2"), arg2 = "TREAT1")

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