将列列表传递给data.table的CJ作为向量

Question

我有以下代码：

main_cols <- c('num', 'let')
dt <- data.table(num = 1:5, let = letters[1:5])
dt

new_dt <- dt[CJ(num = num
                , let = let
                , unique = TRUE)
             , on = main_cols
             ]
head(new_dt, 10)

问题是:我想作为向量传递要交叉连接的列。如何在main_cols函数中“解压”CJ？谢谢。

Answer 1

我认为您应该使用do.call，正如@AnandaMahto建议的那样：

m = dt[, do.call(CJ, .SD), .SDcols=main_cols]
dt[m, on=main_cols]

您还可以以这种方式创建m：

m = do.call(CJ, dt[,main_cols,with=FALSE])

如果列中有重复值，请使用unique选项以CJ

m = dt[, do.call(CJ, c(.SD, unique=TRUE)), .SDcols=main_cols]
# or 
m = do.call(CJ, c(dt[,main_cols,with=FALSE], unique=TRUE))