创建所有交叉组合,尊重列层次结构
从我的data.table DT.in,我想创建所有交叉连接组合,但不仅CJ(_all_my_var_).
我需要执行这样的事情:
CJ(var1)
CJ(var1, var2)
CJ(var1, var2, var3)
...
这是我的代表:
library(data.table)
set.seed(999)
DT.in <- data.table(lvl1 = rep('AA', 200),
code = sample(c('D44', 'J21'), 200, replace = TRUE),
var = sample(c('Z3R', 'TR5', 'JKL', 'FR5', 'TFX'), 200, replace = TRUE),
test = sample(c('ONE', 'TWO', 'THREE', 'FOUR', 'FIVE', 'SIX', 'SEVEN'), 200, replace = TRUE))
foo.1 <- DT.in[, .(new = CJ(lvl1, unique = TRUE))]
foo.2 <- DT.in[, .(new = CJ(lvl1, code, unique = TRUE))]
foo.3 <- DT.in[, .(new = CJ(lvl1, code, var, unique = TRUE))]
foo.4 <- DT.in[, .(new = CJ(lvl1, code, var, test, unique = TRUE))]
DT.out <- rbindlist(list(foo.1, foo.2, foo.3, foo.4), fill = TRUE)
我的DT.out是我想要的,但当然,我需要自动执行此操作,变量的名称和数量永远不会相同。而我还是做不到...
任何帮助做到这一点将不胜感激。
(也许还有另一种方式可以从 DT.in 转到 DT.out ......?)
回答
我们可以使用do.call与CJ
- 循环遍历 'DT.in' 的列序列
lapply - 获取
sequence,根据数字索引选择列 - 应用
CJ在do.call - 将 中的输出数据集绑定
list到一个rbindlist
library(data.table)
DT.out2 <- rbindlist(lapply(seq_along(DT.in), function(i)
do.call(CJ, c(DT.in[, seq(i), with = FALSE], unique = TRUE))), fill = TRUE)
setnames(DT.out2, paste0("new.", names(DT.out2)))
- 检查 OP 的输出
identical(DT.out, DT.out2)
[1] TRUE