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

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

用R中的NA重新编码变量

3 个答案: