如何用if语句或其他方式无报错静默退出R函数？

需求 #

• 函数在满足特定条件时，需像for循环中的break语句一样静默退出，不报错、不触发调试事件
• 函数需具备类似mean()的通用性，可用于dplyr操作及for循环场景
• 避免编写复杂的嵌套if-else语句

现有代码及问题 #

以下是尝试实现的函数，但break/next在函数内无法使用，stop()会触发调试，均不符合需求：

v <- c(1, 2, 5, 1, 2)

break5 <- function(x){
    # 若满足该条件，需退出整个函数
    if (x == 5){
        # next      # <- 执行报错
        break     # <- 执行报错
        # stop()  # <- 触发调试事件，不符合静默需求
    } 
    
    # 否则执行后续逻辑
    if (x == 1) print(1)
    if (x == 2) print(2)
}

控制台执行结果 #

> break5(1)
[1] 1

> break5(2)
[1] 2

> break5(5)
Error in break5(5) : no loop for break/next, jumping to top level

期望效果 #

实现类似for循环中break的静默退出逻辑，例如：

for (i in v){
    if (i == 5) break # <- 希望在函数内部实现该逻辑
    break5(i)
}

示例场景 #

示例1：多条件判断的函数 #

期望函数在不同场景下按需静默退出或正常返回：

x0 <- 1:100
y0 <- 1:100

example <- function(x = NULL, y = NULL){
  
  if (is.null(x) | is.null(y)){    
    break                            # 场景1：静默退出
  }
  
  if (!(x %in% x0) | !(y %in% y0)){
    break                            # 场景2：静默退出
  }
  
  if (x < 2){
    z <- 1                           # 场景3：正常返回
    return(z)
  }
  
  if ((x >= 80) & (x <= 100)){
    break                            # 场景4：静默退出
  }
  
  z <- x + y                        # 其他情况：正常返回
  return(z)
}

example(1)    # 场景1：需静默
example(30, 999)    # 场景2：需静默
example(1, 1)    # 场景3：正常执行
example(80, 1)    # 场景4：需静默
example(11, 1)    # 其他情况：正常执行

示例2：for循环中的调用 #

期望满足条件的迭代被静默处理，不产生无效输出：

for (i in 75:90){
  print(paste0(i, '+', i, '=', example(i, i)))  # 满足条件的迭代需静默处理
}

解决方案 #

在R中，要实现函数的静默退出，直接使用return(invisible(NULL))即可——它会让函数立即终止，且返回的NULL不会在控制台自动打印，完美匹配“静默退出”的需求。

修改后的break5函数 #

break5 <- function(x){
    if (x == 5){
        return(invisible(NULL))  # 静默退出，返回不可见的NULL
    } 
    
    if (x == 1) print(1)
    if (x == 2) print(2)
}

测试结果：

> break5(1)
[1] 1
> break5(2)
[1] 2
> break5(5)
# 无任何输出，静默退出

修改后的example函数 #

x0 <- 1:100
y0 <- 1:100

example <- function(x = NULL, y = NULL){
  
  if (is.null(x) | is.null(y)){    
    return(invisible(NULL))  # 场景1：静默退出
  }
  
  if (!(x %in% x0) | !(y %in% y0)){
    return(invisible(NULL))  # 场景2：静默退出
  }
  
  if (x < 2){
    z <- 1                           # 场景3：正常返回
    return(z)
  }
  
  if ((x >= 80) & (x <= 100)){
    return(invisible(NULL))  # 场景4：静默退出
  }
  
  z <- x + y                        # 其他情况：正常返回
  return(z)
}

测试各场景：

> example(1)
# 无输出（场景1：静默退出）
> example(30, 999)
# 无输出（场景2：静默退出）
> example(1, 1)
[1] 1
> example(80, 1)
# 无输出（场景4：静默退出）
> example(11, 1)
[1] 12

示例2的优化调用 #

为避免for循环中打印NULL，可先判断返回值是否有效：

for (i in 75:90){
  res <- example(i, i)
  if (!is.null(res)) {
    print(paste0(i, '+', i, '=', res))
  }
}

执行后仅会打印75-79的计算结果，80-90的迭代因函数静默退出而无输出，符合预期。

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