Условная замена на NA в R (два кадра данных) - PullRequest
Новая
       52

Условная замена на NA в R (два кадра данных)

0 голосов
/ 02 июля 2018

у меня 

    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 с правильными индексами, но сокращает id до длины idx.

1 Ответ

0 голосов
/ 02 июля 2018

Я хочу, чтобы 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] - это объединение, в котором строки mDT ищутся в DT с использованием on=
  • j выполняется в объединенном подмножестве DT
  • когда j равно k := expr, столбец k из DT обновлен
...