data.table 中的函数，具有两列作为参数

Question

我有以下功能：

DT <- data.table(col1 = 1:4, col2 = c(2:5))

fun <- function(DT, fct){
  DT_out <- DT[,new_col := fct]
  return(DT_out)
}

fun(input, fct = function(x = col1, y = col2){y - x})

实际上我在这段代码sn-p之前和之后都有一些处理，因此我不希望直接使用带有固定fct的语句DT[,new_col := fct]（因为fct应该是灵活的）。我知道这个问题与this 非常相似，但我无法弄清楚如何重新编写代码，以便允许将两列作为函数的参数。上面的代码给出了错误：

Error in `[.data.table`(DT, , `:=`(new_col, fct)) : 
  RHS of assignment is not NULL, not an an atomic vector (see ?is.atomic) and not a list column. 

Answer 1

如果您不介意在变量名周围添加引号，则可以选择一个

fun <- function(DT, fun, ...){
  fun_args <- c(...)
  DT[,new_col := do.call(fun, setNames(mget(fun_args), names(fun_args)))]
}

fun(DT, fun = function(x, y){y - x}, x = 'col1', y = 'col2')

DT
#    col1 col2 new_col
# 1:    1    2       1
# 2:    2    3       1
# 3:    3    4       1
# 4:    4    5       1

或者使用.SDcols（结果同上）

fun <- function(DT, fun, ...){
  fun_args <- c(...)
  DT[, new_col := do.call(fun, setNames(.SD, names(fun_args))), 
     .SDcols = fun_args]
}