创建所有交叉组合,尊重列层次结构

从我的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.callCJ

  1. 循环遍历 'DT.in' 的列序列 lapply
  2. 获取sequence,根据数字索引选择列
  3. 应用CJdo.call
  4. 将 中的输出数据集绑定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


以上是创建所有交叉组合,尊重列层次结构的全部内容。
THE END
分享
二维码
< <上一篇
下一篇>>