是否可以在 data.table 中以 R 中的变量形式在参数上传递列名

Question

我正在尝试对 data.table 执行非 equi 连接，如下所示：

A <- data.table(p = c(10, 20, 30), q = c(15, 25, 35), r = c(0, 0, 0))
B <- data.table(x = c(5, 15), y = c(15, 25), z = c(1, 2))

这行得通：

A[B, on = .(p >= x, p < y), r := r + z]

这不起作用：

col_var <- "p"
A[B, on = .(col_var >= x, col_var < y), r := r + z]
or
A[B, on = .(.(col_var) >= x, .(col_var) < y), r := r + z]

我正在尝试实现这样的目标：

col_var <- c("p", "q")
for(i in col_var)
{
  A[B, on = .(.(i) >= x, .(i) < y), r := r + z]
}

当我直接使用 p 而不是 col_var 时，我能够得到结果，但我需要使用 for 循环，这需要我将列名存储在变量中。有没有办法可以使用存储在变量中的列名？

PS：我尝试使用 .(col_var) 但我收到错误“参数指定列指定不存在的列”

Answer 1

您可以在on 参数中使用字符串。这允许我们使用paste 构造连接：

A[B, on = paste(col_var, c(">= x", "< y")), r := r + z]