R语言函数2输入3时报错Error in if (x[x<0]) { : argument is of length zero的原因

问题重现 #

你编写的两个函数如下：

# function 1
abs.x <- function(x){
  if (x<0) {x=-x}
  x
}

# function 2
abs.x <- function(x){  
  if(x[x<0]){
     x[x<0]=-x[x<0]
  }
  x
}

测试函数2时，输入3触发报错：

abs.x(3)
[1] Error in if (x[x<0]) { : argument is of length zero

但输入-3和c(3,-3)时结果正常。

报错原因解析 #

核心问题出在函数2的if(x[x<0])条件判断上：

• 当输入3时，x<0的结果是FALSE，x[x<0]会提取空向量（长度为0）。而R的if语句要求条件必须是长度为1的逻辑值，空向量不符合要求，直接触发报错。
• 当输入-3时，x<0是TRUE，x[x<0]得到-3。R会自动将非零数值转换为逻辑值（非零等价于TRUE），所以if(-3)等价于if(TRUE)，条件成立，执行赋值操作将-3改为3，结果正确。
• 当输入c(3,-3)时，x[x<0]得到-3，同样被转为TRUE，条件成立，修改-3为3，输出正确。

简单修复方案 #

其实不需要用if判断，直接赋值即可避免报错：

abs.x <- function(x){
  x[x < 0] <- -x[x < 0]
  x
}

没有符合x<0的元素时，赋值操作会自动跳过，不会改变原向量。

内容的提问来源于stack exchange，提问作者Lulu