R: ifelse превращает число c (0) в NA

Question

Может кто-нибудь объяснить мне, почему происходит следующее?

ifelse(TRUE, numeric(0), 1)
> [1] NA

Я бы, конечно, ожидал цифру c (0). Я подозреваю, что это потому, что ifelse векторизован, например, следующие работы, но я не совсем понимаю, что происходит.

if (TRUE) numeric(0) else 1
#> numeric(0)

Answer 1

вы можете получить доступ к реализации ifelse, которая

function (test, yes, no) 
{
    if (is.atomic(test)) {
        if (typeof(test) != "logical") 
            storage.mode(test) <- "logical"
        if (length(test) == 1 && is.null(attributes(test))) {
           #... let's skip this part..
        }
    }
    else test <- if (isS4(test)) 
        methods::as(test, "logical")
    else as.logical(test)
    ans <- test
    len <- length(ans)
    ypos <- which(test)
    npos <- which(!test)
    if (length(ypos) > 0L) 
        ans[ypos] <- rep(yes, length.out = len)[ypos]
    if (length(npos) > 0L) 
        ans[npos] <- rep(no, length.out = len)[npos]
    ans
}
<bytecode: 0x00000123e6b7d3a0>
<environment: namespace:base>

Итак, да, это потому, что ifelse векторизовано - в частности, по условию - и возвращаемый объект ans инициализируется как вектор такой же длины, что и условие.

Описание ifelse состояний

ifelse возвращает значение с той же формой, что и у test который заполняется элементами, выбранными из да или нет, в зависимости от того, является ли элемент теста ИСТИНА или ЛОЖЬ.

Пусть test <- TRUE. Интересные строки

ypos <- which(test)
rep(numeric(0), length.out = 1)[ypos]

Answer 2

Если вы хотите настроить функцию так, чтобы она возвращала numeric(0) в вашей ситуации, вы можете изменить if(length(yes) == 1) на if (length(yes) == 0 | length(yes) == 1) внутри функции. Это дает вам:

ifelse2 <- function (test, yes, no) {
  if (is.atomic(test)) {
    if (typeof(test) != "logical") 
      storage.mode(test) <- "logical"
    if (length(test) == 1 && is.null(attributes(test))) {
      if (is.na(test)) 
        return(NA)
      else if (test) {
    if (length(yes) == 0 | length(yes) == 1) { # Here is what I changed
          yat <- attributes(yes)
          if (is.null(yat) || (is.function(yes) && identical(names(yat), 
                                                         "srcref"))) 
            return(yes)
        }
  }
      else if (length(no) == 1) {
        nat <- attributes(no)
        if (is.null(nat) || (is.function(no) && identical(names(nat), 
                                                          "srcref"))) 
          return(no)
      }
    }
  }
  else test <- if (isS4(test)) 
    methods::as(test, "logical")
  else as.logical(test)
  ans <- test
  len <- length(ans)
  ypos <- which(test)
  npos <- which(!test)
  if (length(ypos) > 0L) 
    ans[ypos] <- rep(yes, length.out = len)[ypos]
  if (length(npos) > 0L) 
    ans[npos] <- rep(no, length.out = len)[npos]
  ans
}

Попытка:

ifelse2(TRUE, numeric(0), 1)
> [1] numeric(0)

Вы можете сделать то же самое с аргументом no, если в вашем случае это тоже может быть numeric(0).