在R语言中使用segmented包时如何跳过错误并统计错误次数

原代码存在两个核心问题：一是函数名拼写错误（selgmented应为segmented）；二是当无结构断点或牛顿法收敛失败时，函数抛出错误导致循环中断。以下是修改后的代码，实现错误捕获、跳过错误迭代并统计错误次数的功能：

library(segmented)

iteration <- 1
max_iter <- 1001
a <- matrix(0, max_iter, 1)
b <- matrix(0, max_iter, 1)
c <- matrix(0, max_iter, 1)
error_count <- 0  # 新增错误计数变量

repeat{
  TS <- cumsum(rnorm(100, 0, 1))
  idt <- 1:length(TS)
  o1 <- lm(TS ~ idt)
  
  # 使用tryCatch捕获错误
  os1_result <- tryCatch(
    expr = segmented(o1, Kmax = 2, type = "bic"),
    error = function(e) {
      # 出错时执行的逻辑，可选打印错误信息
      # message(paste("迭代", iteration, "出错：", e$message))
      return(NULL)  # 返回NULL标记本次迭代出错
    }
  )
  
  if (!is.null(os1_result)) {
    # 执行成功时统计断点次数
    n_breaks <- length(os1_result$psi[, 1])
    a[iteration] <- ifelse(n_breaks == 1, 1, 0)
    b[iteration] <- ifelse(n_breaks == 2, 1, 0)
    c[iteration] <- ifelse(n_breaks == 0, 1, 0)
  } else {
    # 出错时错误计数+1，对应行设为NA便于后续识别
    error_count <- error_count + 1
    a[iteration] <- NA
    b[iteration] <- NA
    c[iteration] <- NA
  }
  
  iteration <- iteration + 1
  if (iteration > max_iter) break
}

# 输出最终错误统计结果
cat("总错误次数：", error_count, "\n")

关键修改说明 #

• 修正函数名：将拼写错误的selgmented改为官方正确的segmented，避免基础语法错误。
• 错误捕获机制：用tryCatch()包裹segmented调用，捕获所有执行错误，出错时返回NULL并递增错误计数器，保证循环不中断。
• 分支逻辑处理：根据执行结果是否为NULL，分别处理成功迭代（统计断点次数）和错误迭代（记录错误）的情况。
• 错误统计：新增error_count变量跟踪错误总次数，最后可直接输出统计结果。

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