data.table 使用常量更新连接

Question

我想根据联接和固定值更新 data.table 中的行子集。

d1 <- data.table(A = 5:1, B = letters[5:1])    
d2 <- data.table(C = letters[5:1], Z = 6:10)
current.val <- 5

我想要做的是根据与 d2 的连接更新 d1，但仅限于 d1 中 A==5 的位置。像这样：

d1[d2, D := i.Z ,on=.(B==C, A==current.val)]

我目前的方法是向 d2 添加一个新列并将其设置为固定值并在连接中使用它：

d2[, current.val := 5]
d1[d2, D := i.Z ,on=.(B==C, A==current.val)]

这可行，但似乎开销很大。有没有更简单的方法在连接中使用常量值？

---

(8/14) 基准测试的新比例示例：

d1 <- data.table(A = 100:1, B = 100000000:1, D = as.numeric(NA),  key = c("A", "B"))
d2 <- data.table(C = 100000000:1, Z = c(10:1) / 10, key = "C")
current.val <- 5

system.time(d1[cbind(d2, A = current.val), on = .(B = C, A), D := i.Z])
system.time({setkey(d1, B, A); d1[d1[d2][A == current.val], D := Z]; setkey(d1, A, B)})
system.time(d1[d1[d2][A == current.val], D := Z]) # fastest, if inverse key order is acceptable

Answer 1

我目前的方法是向 d2 添加一个新列并将其设置为固定值并在连接中使用它[...] 有没有更简单的方法在连接中使用常量值？

这是一个好方法。或者，您可以使用 cbind 在联接中临时添加一列：

d1[cbind(d2, A = current.val), on=.(B = C, A), D := i.Z ]

实际上，c 在这里可以代替 cbind，但我觉得这是一种更奇怪的方法。

Answer 2

只有五行，它不会有任何区别，但如果你想扩大，那么如果你在d1和d2上设置键，这个操作会快得多：

d1 <- data.table(A=5:1, B=letters[5:1], key="B")
d2 <- data.table(C=letters[5:1], Z=6:10, key="C")

然后，以下构造将给出所需的结果：

d1[d1[d2][A==current.val], D := Z]