在R中用NA进行条件替换(两个数据帧)

时间:2018-07-02 18:38:47

标签: r replace conditional na

我有

    idx <- c(1397, 2000, 3409, 3415, 4077, 4445, 5021, 5155) 

    idy <- c( 1397, 2000, 2860, 3029, 3415, 3707, 4077, 4445, 5021, 5155, 
             5251, 5560)

   agex <- c(NA, NA, NA, 35, NA, 62, 35, 46)

   agey <- c( 3, 45,  0, 89,  7,  2, 13, 24, 58,  8,  3, 45)


   dat1 <- as.data.frame(cbind(idx, agex))
   dat2 <- as.data.frame(cbind(idy, agey))

现在我想要每当agex = NA,且idx = idy时,agey = NA,所以

       idy agey
  1    1397   NA
  2    2000   NA
  3    2860    0
  4    3029   89
  5    3415    7
  6    3707    2
  7    4077   NA
  8    4445   24
  9    5021   58
  10   5155    8
  11   5251    3
  12   5560   45

我已经尝试过了

ifelse(is.na(dat1$agex) | dat1$idx %in% dat2$idy, NA, dat2$agey)

它以正确的索引返回NA,但是将idy缩短为idx的长度。

1 个答案:

答案 0 :(得分:1)

  

我想每当agex = NA,idx = idy时,agey = NA

使用data.table更新联接...

library(data.table)
setDT(dat1); setDT(dat2)

dat2[dat1[is.na(agex)], on=.(idy = idx), agey := NA]

dat2

     idy agey
 1: 1397   NA
 2: 2000   NA
 3: 2860    0
 4: 3029   89
 5: 3415    7
 6: 3707    2
 7: 4077   NA
 8: 4445   24
 9: 5021   58
10: 5155    8
11: 5251    3
12: 5560   45

工作原理

  • dat1[is.na(agex)]agex为NA
  • 的子集
  • DT[mDT, on=, j]是一个联接,其中使用mDTDT中查找on=的行
  • jDT的连接子集中完成
  • jk := expr时,k的列DT被更新