用data.table将表的一半与笛卡尔连接

Question

假设我有以下data.table：

library(data.table)
dt <- data.table(
  a = c(1L, 1L, 2L, 2L), 
  b = c('A', 'A', NA_character_, NA_character_),
  Amount = 1:4
)

我想执行笛卡尔联接以将权重应用于“数量”列，例如我有下表的权重：

set.seed(42L)
weights <- data.table(
  a = c(rep(1L, 5L), rep(2L, 5L)),
  b = LETTERS[1:10],
  Weight = runif(10L)
)

一种获得我想要的方法是：

rbind(
  dt[a == 1L],
  dt[a != 1L][
    weights,
    .(a, b = i.b, Amount = Amount * Weight),
    on = c('a'),
    nomatch = 0L,
    allow.cartesian = TRUE
  ]
)

编辑：我同事的一项改进，这是对第一项的改进，但对性能没有太大的作用：

rbind(
  dt[a == 1L],
  dt[a != 1L][
    weights,
    modifyList(.SD, .(b = i.b, Amount = Amount * Weight)),
    on = c('a'),
    nomatch = 0L,
    allow.cartesian = TRUE
    ]
)

导致：

    a b    Amount
 1: 1 A 1.0000000
 2: 1 A 2.0000000
 3: 2 F 1.5572878
 4: 2 F 2.0763838
 5: 2 G 2.2097649
 6: 2 G 2.9463533
 7: 2 H 0.4039998
 8: 2 H 0.5386664
 9: 2 I 1.9709769
10: 2 I 2.6279692
11: 2 J 2.1151944
12: 2 J 2.8202591

我想知道/改善两件事：

1. 如果列数增加，则需要在代码的最后一块中指定所有列.(a, b = ib, Amount = Amount * Weight)持续增长。
2. 如果行数增加到数百万，有没有办法更有效地做到这一点？

Answer 1

对于qn1，可能类似于：

othercols <- setdiff(names(weights), "Weight")
rbindlist(list(
    dt[a == 1L],
    dt[a != 1L][
        weights,
        c(setNames(mget(paste0("i.", othercols)), othercols), .(Amount = Amount * Weight)),
        on=.(a),
        nomatch = 0L,
        allow.cartesian = TRUE
    ]
), use.names=TRUE)

对于qn2，是否真的需要笛卡尔联接来将数据扩展这么大？