连接data.frame或data.table中的匹配列

时间:2012-03-29 02:46:18

标签: r merge dataframe plyr data.table

我有以下data.frames:

a <- data.frame(id = 1:3, v1 = c('a', NA, NA), v2 = c(NA, 'b', 'c'))
b <- data.frame(id = 1:3, v1 = c(NA, 'B', 'C'), v2 = c("A", NA, NA))
> a
  id   v1   v2
1  1    a <NA>
2  2 <NA>    b
3  3 <NA>    c
> b
  id   v1   v2
1  1 <NA>    A
2  2    B <NA>
3  3    C <NA>

注意:两个表中都没有定义v1或v2的ID;每个id值

在每列中只有一个唯一的非NA值

我想将这些数据框合并到“id':

的匹配值上
ab <- merge(a, b, by = "id")

但我还希望合并两列v1v2,以便data.frame ab如下所示:

ab <- data.frame(id = 1:3, v1 = c("a", "B", "C"), v2 = c("A", "b", "c"))

> ab
  id v1 v2
1  1  a  A
2  2  B  b
3  3  C  c

相反,我明白了:

> merge(a, b, by = "id")
  id v1.x v2.x v1.y v2.y
1  1    a <NA> <NA>    A
2  2 <NA>    b    B <NA>
3  3 <NA>    c    C <NA>

使用data.framedata.table的示例会很有帮助,所以这里是上面的data.table版本:

A <- data.table(a, key = 'id')
B <- data.table(b, key = 'id')
A[B]

2 个答案:

答案 0 :(得分:7)

使用merge(带数据框)可能无法指定您指定的合并类型,尽管说通常会被证明是错误的。

您还省略了一些细节:每个NA值的每列中是否总会有一个唯一的非id值?如果是这样,这将有效:

ab <- rbind(a,b)
> colFun <- function(x){x[which(!is.na(x))]}
> ddply(ab,.(id),function(x){colwise(colFun)(x)})
  id v1 v2
1  1  a  A
2  2  B  b
3  3  C  c

类似的策略也适用于data.table

abDT <- data.table(ab,key = "id")
> abDT[,list(colFun(v1),colFun(v2)),by = id]
     id V1 V2
[1,]  1  a  A
[2,]  2  B  b
[3,]  3  C  c

答案 1 :(得分:4)

如果您的数据非常简单,那么joran的回答可能是最简单的方法。这可能是基础:

a <- data.frame(id = 1:3, v1 = c('a', NA, NA), v2 = c(NA, 'b', 'c'))
b <- data.frame(id = 1:3, v1 = c(NA, 'B', 'C'), v2 = c("A", NA, NA))

decider <- function(x, y) factor(ifelse(is.na(x), as.character(y), as.character(x)))
data.frame(mapply(a, b, FUN = decider))

如果您的数据有不同的ID(有些重叠,有些则没有,那么这里有不同的方法:

a <- data.frame(id = c(1,2,4,5), v1 = c('a', NA, "q", NA), v2 = c(NA, 'b', 'c', "e"))
b <- data.frame(id = 1:4, v1 = c(NA, "A", "C", 'B'), v2 = c("A", NA, "D", NA))

decider <- function(x, y) factor(ifelse(is.na(x), as.character(y), as.character(x)))

DF <- data.frame(mapply(a, b, FUN = decider))
DF2 <- rbind(b[!b$id %in% DF$id , ], DF)
DF2 <- DF2[order(DF2$id), ]
rownames(DF2) <- 1:nrow(DF2)