当A列的值匹配时,将B列的NA替换为B列的值

时间:2018-09-18 15:13:03

标签: r merge pattern-matching na

我正在尝试基于两个变量(Entrez.ID和Gene.ID)合并两个数据框。一个数据帧只有这些变量,例如

Entrez.ID  Gene.ID
10007      GNPDA1
10016      ALG2
10044      SH2D3C 

和一个包含三个变量的数据框,例如

Entrez.ID    Gene.ID   Ensembl.ID
10007        GPI       ENSG00000113552
10016        PDCD6     ENSG00000249915
10044        CHAT      ENSG00000095370

当前,当我使用以下方式合并文件时:

df<-merge(df1,df2,by=c("Entrez.ID","Gene.ID"),all=TRUE)

我得到一个看起来像这样的数据框:

Entrez.ID   Gene.ID   Ensembl.ID
10007       GNPDA1    <NA>
10007       GPI       ENSG00000113552
10016       ALG2      <NA>
10016       PDCD6     ENSG00000249915
10044       SH2D3C    <NA>
10044       CHAT      ENSG00000095370

但是我要创建的是一个看起来像这样的数据框:

Entrez.ID   Gene.ID   Ensembl.ID
10007       GNPDA1    ENSG00000113552
10007       GPI       ENSG00000113552
10016       ALG2      ENSG00000249915
10016       PDCD6     ENSG00000249915
10044       SH2D3C    ENSG00000095370
10044       CHAT      ENSG00000095370

如何告诉R Entrez.ID变量在哪里匹配,我希望Ensembl.ID匹配(即在可用的情况下用Ensembl.ID替换NA)?

2 个答案:

答案 0 :(得分:1)

我们可以使用na.locf中的zoo

library(zoo)
df$Ensembl.ID <- with(df, ave(Ensembl.ID, Entrez.ID, FUN = function(x)
        na.locf(na.locf(x, na.rm = FALSE), fromLast = TRUE)))
df$Ensembl.ID
#[1] "ENSG00000113552" "ENSG00000113552" "ENSG00000249915" 
#[4] "ENSG00000249915" "ENSG00000095370"
#[6] "ENSG00000095370"

或使用full_join中的dplyr

library(tidyverse)
full_join(df1, df2, by = c("Entrez.ID","Gene.ID")) %>%
    group_by(Entrez.ID) %>%
    fill(Ensembl.ID, .direction = 'up') %>%
    fill(Ensembl.ID, .direction = 'down')
# A tibble: 6 x 3
# Groups:   Entrez.ID [3]
#  Entrez.ID Gene.ID Ensembl.ID     
#      <int> <chr>   <chr>          
#1     10007 GNPDA1  ENSG00000113552
#2     10007 GPI     ENSG00000113552
#3     10016 ALG2    ENSG00000249915
#4     10016 PDCD6   ENSG00000249915
#5     10044 SH2D3C  ENSG00000095370
#6     10044 CHAT    ENSG00000095370

数据

df <- structure(list(Entrez.ID = c(10007L, 10007L, 10016L, 10016L, 
10044L, 10044L), Gene.ID = c("GNPDA1", "GPI", "ALG2", "PDCD6", 
"SH2D3C", "CHAT"), Ensembl.ID = c(NA, "ENSG00000113552", NA, 
"ENSG00000249915", NA, "ENSG00000095370")), class = "data.frame", 
 row.names = c(NA, -6L))

答案 1 :(得分:0)

如果df1的Gene.ID和df2的Gene.ID是唯一的(df1中的ID不是df2中的ID),则可以简单地合并

df <- merge(df1,df2,by=c("Entrez.ID"),all.x=TRUE)

然后使用rbind()

df <- rbind(df, df2)

PS。我建议将来使用data.table进行数据整理。它非常直观且速度更快。