R语言使用tryCatch语句编写向量平方根计算函数的问题

现有代码问题说明 #

• 第一版代码缺陷：if(x < 0)仅对长度为1的输入生效，当输入是多元素向量时，只会判断第一个元素的正负，逻辑存在漏洞；同时缺少返回值定义，也未接入tryCatch异常捕获逻辑。
• 第二版代码缺陷：核心计算逻辑sqrt(any(x))完全不符合需求，any(x)返回的是逻辑值（判断x中是否存在非0值），转换为数值后仅为1或0，无法得到向量x的平方根结果；warning处理函数也没有按要求返回NaN，逻辑不匹配。

tryCatch放置位置说明 #

tryCatch需要包裹住核心的平方根计算逻辑段，确保计算过程中因负数开方产生的警告可以被正常捕获，你可以直接按照下方正确实现调整代码：

sqrt_x <- function(x){
  tryCatch(
    expr = {
      # 核心计算逻辑：对向量x求平方根
      res <- sqrt(x)
      # 无异常时直接返回计算结果
      return(res)
    },
    warning = function(w){
      # 捕获到负数开方的警告时，抛出自定义提示，返回NaN
      warning("x包含负数")
      return(NaN)
    }
  )
}

功能验证示例 #

• 输入非负向量：sqrt_x(c(4, 9, 16))，返回结果为 2 3 4
• 输入含负数的向量：sqrt_x(c(4, -9, 16))，会触发警告警告信息：x包含负数，返回结果为 NaN

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