如果三个列中至少之一重复一个值,则分配唯一ID

时间:2019-05-08 15:05:26

标签: r

我有一个包含三列的数据框。这些列上有许多缺失值。我想创建一个新的唯一ID列,为每一行提供相同的ID,这三列中的至少一列具有相同的值。

数据框看起来像这样

df=data.frame(id1=c(5,5,NA,NA,NA,5,4,NA,NA),
              id2=c(NA,58,58,58,NA,58,NA,66,66),
              id3=c(NA,NA,NA,581,581,581,NA,669,NA))
df
  id1 id2 id3
1   5  NA  NA
2   5  58  NA
3  NA  58  NA
4  NA  58 581
5  NA  NA 581
6   5  58 581
7   4  NA  NA
8  NA  66 669
9  NA  66  NA

新列应如下所示

df=data.frame(id1=c(5,5,NA,NA,NA,5,4,NA,NA),
              id2=c(NA,58,58,58,NA,58,NA,66,66),
              id3=c(NA,NA,NA,581,581,581,NA,669,NA),
              ID=c(1,1,1,1,1,1,2,3,3))
  id1 id2 id3 ID
1   5  NA  NA  1
2   5  58  NA  1
3  NA  58  NA  1
4  NA  58 581  1
5  NA  NA 581  1
6   5  58 581  1
7   4  NA  NA  2
8  NA  66 669  3
9  NA  66  NA  3

1 个答案:

答案 0 :(得分:0)

鉴于您对@Gregor的问题的回答,这是一个可以概括的解决方案:

ID1.criteria <- c(5,58,581)
ID2.criteria <- c(4)
ID3.criteria <- c(66,669)
ID <- NULL
for (i in 1:nrow(df)) {
  if (any(df[i,] %in% ID1.criteria)) {
    ID[i] <- 1
  } else {
    if (any(df[i,] %in% ID2.criteria)) {
      ID[i] <- 2
    } else {
      if (any(df[i,] %in% ID3.criteria)) {
        ID[i] <- 3
      } else {
        ID[i] <- NA
      }
    }
  }
}
df$ID <- ID
df
  id1 id2 id3 ID
1   5  NA  NA  1
2   5  58  NA  1
3  NA  58  NA  1
4  NA  58 581  1
5  NA  NA 581  1
6   5  58 581  1
7   4  NA  NA  2
8  NA  66 669  3
9  NA  66  NA  3

如果需要,您可以将条件设​​置为所需的条件,因为有限的一组id1,id2和id3对应于相同的ID号,而与它们的出现顺序无关,并且它们永远不会在同一行上。