r - 在 R 中使用 NA 重新编码变量

Question

我正在尝试根据其他三个变量的值编写一个新变量。具体来说，如果所有变量都是 NA，我希望新变量取 NA，如果其中任何一个为 1，则应取 1，否则应取 0。但是，使用以下代码时遇到错误：

frame <- data.frame(x = c(NA,NA,1), x2 = c(NA, NA, 0), x3 = c(NA,0,0))
frame

y <- ifelse(is.na(frame$x) == TRUE & is.na(frame$x2) == TRUE & is.na(frame$x3) == TRUE, NA, 0)
y2 <- ifelse(frame$x == 1 | frame$x2 == 1 | frame$x3 == 1, 1, y)

y 的第二个数字正确为 0，但在 y2 中引用时变为 NA。知道为什么会这样吗？

Answer 1

获得预期输出的一种选择是

  do.call(pmax, c(frame, na.rm=TRUE))
  #[1] NA  0  1

Answer 2

你有三个不同的条件，所以用三行来表达是最自然的：

z <- rep(0,nrow(frame))

z[apply(is.na(frame),1,all)] <- NA
z[apply(frame==1    ,1,any)] <- 1
# [1] NA  0  1

Answer 3

如果您有这样的 data.frame：

frame <- data.frame(a=letters[1:5], x1=c(1,1,NA,NA,0), x2=c(1,0,NA,0,0), x3=c(0,1,NA,1,0))

> frame
  a x1 x2 x3
1 a  1  1  0
2 b  1  0  1
3 c NA NA  NA
4 d NA  0  1
5 e  0  0  0

像这样的事情可能会做你所追求的：

frame$summary <- apply(frame[,c('x1','x2','x3')], 1, function(row) {
    if (all(is.na(row))) {
        return(NA)
    } else if (1 %in% row) {
        return(1)
    } else {
      return(0)
    }})

> frame
  a x1 x2 x3 summary
1 a  1  1  0       1
2 b  1  0  1       1
3 c NA NA NA      NA
4 d NA  0  1       1
5 e  0  0  0       0

...并且是将您的口头描述相当直接地翻译成代码。