R语言data.table中如何在pmax()内传入动态列变量？

你当前写法无效的本质是：substitute(example_column)只会捕获变量example_column本身的符号引用，不会将它存储的字符串值转换为data.table数据环境内的列对象，等价于直接把字符串常量传入pmax()，自然无法匹配到对应列的数值。

方案1：基础取列法（单/多动态列通用） #

data.table的计算环境内，get(列名字符串)可以直接取出对应列的向量：

• 单个动态列场景：

DT <- as.data.table(iris) 
example_column <- c("Petal.Width")
DT[, test_flag :=
     ifelse(pmax(get(example_column), Petal.Length) > 0, "Exclude", "Consider")]

• 多个动态列场景：如果动态列是存储在字符向量里的一组列名，用mget()批量取列后，通过do.call()把参数传给pmax()即可：

# 动态列集合示例
example_columns <- c("Petal.Width", "Sepal.Width")
DT[, test_flag :=
     ifelse(do.call(pmax, c(mget(example_columns), Petal.Length)) > 0, "Exclude", "Consider")]

方案2：data.table 规范写法（配合.SDcols） #

用data.table内置的.SDcols参数指定需要动态引用的列，再通过.SD引用子集数据，可读性更强，也符合data.table的语法设计：

example_column <- c("Petal.Width")
DT[, test_flag :=
     ifelse(pmax(.SD[[1]], Petal.Length) > 0, "Exclude", "Consider"),
   .SDcols = example_column] # 把动态列名传给.SDcols

如果是多动态列，直接把列名字符向量传给.SDcols，.SD会自动包含所有指定的动态列，可直接作为pmax()的参数：

example_columns <- c("Petal.Width", "Sepal.Width")
DT[, test_flag :=
     ifelse(pmax(.SD, Petal.Length) > 0, "Exclude", "Consider"),
   .SDcols = example_columns]

方案3：rlang 动态语法注入 #

如果项目已经引入rlang生态，可以通过符号转换+表达式注入的方式传动态列：

library(rlang)
col_sym <- sym(example_column) # 把字符串转成列符号
DT[, test_flag :=
     ifelse(pmax(!!col_sym, Petal.Length) > 0, "Exclude", "Consider")]

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