根据另一个数据表更新一个 data.table

Question

仅当值不为 NA 时，我想根据新的 data.table 更新旧的 data.table 中的列。

DT_old = data.table(x=rep(c("a","b","c")), y=c(1,3,6), v=1:3, l=c(1,1,1))
DT_old
   x y v l
1: a 1 1 1
2: b 3 2 1
3: c 6 3 1
DT_new = data.table(x=rep(c("b","c",'d')), y=c(9,6,10), v=c(2,NA,10), z=c(9,9,9))
DT_new
   x  y  v z
1: b  9  2 9
2: c  6  NA 9
3: d 10 10 9

我希望输出是

   x  y  v z
1: b  9  2 9
2: c  6  3 9
3: d 10 10 9
4: a 1 1 NA

目前我正在合并两个 data.table 并遍历每一列并替换新 data.table 中的 NA

DT_merged <- merge(DT_new, DT_old, all=TRUE, by='x')
DT_merged
   x y.x v.x  z y.y v.y  l
1: a  NA  NA NA   1   1  1
2: b   9   2  9   3   2  1
3: c   6  NA  9   6   3  1
4: d  10  10  9  NA  NA NA
DT_merged[is.na(y.x), y.x := y.y]
DT_merged[is.na(v.x), v.x := v.y]
DT_merged = DT_merged[, list(y=y.x, v=v.x, z=z)

有没有更好的方法来完成上述操作？

Answer 1

以下是我的处理方法。首先我将DT_new根据两个表的x列的唯一值组合使用二元连接展开

res <- setkey(DT_new, x)[unique(c(x, DT_old$x))]
res
#    x  y  v  z
# 1: b  9  2  9
# 2: c  6 NA  9
# 3: d 10 10  9
# 4: a NA NA NA

然后，我将使用另一个二进制连接通过引用更新res 中的两列

setkey(res, x)[DT_old, `:=`(y = i.y, v = i.v)]
res
#    x  y  v  z
# 1: a  1  1 NA
# 2: b  3  2  9
# 3: c  6  3  9
# 4: d 10 10  9

---

在 cmets 部分之后，您似乎正在尝试按自己的条件加入 each 列。在 R 或任何语言 AFAIK 中都没有简单的方法来做这样的事情。因此，您自己的解决方案本身可能是一个不错的选择。

不过，这里还有一些其他的选择，主要取自a similar question I myself asked not long ago

使用两个ifelse 语句

setkey(res, x)[DT_old, `:=`(y = ifelse(is.na(y), i.y, y), 
                            v = ifelse(is.na(v), i.v, v))]

两个独立的条件连接

setkey(res, x) ; setkey(DT_old, x) ## old data set needs to be keyed too now
res[is.na(y), y := DT_old[.SD, y]]
res[is.na(v), v := DT_old[.SD, v]]

两者都能满足你的需要。

---

附言

如果您不想要警告，则需要正确定义相应的列类，例如DT_new 中的v 列应定义为v= c(2L, NA_integer_, 10L)