合并具有两个 id 变量和 NA 值的两个数据表

Question

我要合并以下数据tables

dt1 <- data.table(id = letters[1:5], day = 1, var1 = c(2,5,8,7,9), var2 = c(5,5,8,6,7), key = "id")
dt2 <- data.table(id = letters[3:7], day = 2, var1 = c(1,7,6,6,3), var2 = c(2,3,3,2,1), key = "id")

并且结果应包括每个 ID 的每一天。不幸的是，有些 ID 在几天内不存在。

id day var1 var2
a   1   2   5
a   2   NA  NA
b   1   5   5
b   2   NA  NA
c   1   8   8
c   2   1   2
d   1   7   6
d   2   7   3

我已经尝试将 id 和 day 设置为 DT 的键。对于以下几行，我无法为 id 获得 day 2，因为它实际上丢失了，并且变量加倍 (var1.x var1.y)

merge(dt1, dt2, by= c("id","day"), all=TRUE)
merge(dt1, dt2, by= c("day","id"), all=TRUE)

allow.cartesian也不行。任何人都知道 idea/comment 如何获得我需要的最终 table 吗？

Answer 1

尝试

library(data.table)
dcast(melt(rbind(dt1, dt2), id=c('id', 'day')), 
        id+day~variable, value.var='value', drop=FALSE)
# 1:  a   1    2    5
# 2:  a   2   NA   NA
# 3:  b   1    5    5
# 4:  b   2   NA   NA
# 5:  c   1    8    8
# 6:  c   2    1    2
# 7:  d   1    7    6
# 8:  d   2    7    3
# 9:  e   1    9    7
#10:  e   2    6    3
#11:  f   1   NA   NA
#12:  f   2    6    2
#13:  g   1   NA   NA
#14:  g   2    3    1

或者正如@BramVisser 评论的那样，将 rbind(dt1, dt2) 替换为 rbindlist(list(dt1, dt2))

或另一种不使用 melt/dcast

的方法

rbindlist(list(dt1, dt2))[, if(.N <2) .SD[c(.N, .N+1)] else .SD, id][,
    day:=replace(day, is.na(day), setdiff(1:2,na.omit(day))) , id][]

或

setkey(rbindlist(list(dt1, dt2)), id, day)[CJ(id=unique(id), day=unique(day))]

合并具有两个 id 变量和 NA 值的两个数据表

Merging two data tables with two id variables and NA values

merge

join

r

data.table