在R中如何捕获ggplot函数运行产生的警告信息并存入变量

完全可以通过基础R的原生警告处理函数实现，核心使用withCallingHandlers()捕获指定警告，无需加载额外第三方包。

优化后的函数代码 #

draw_plot <- function(data, ...){
  # 初始化变量存储捕获到的警告信息
  caught_warn <- NULL
  
  plot <- ggplot(data = data, mapping = aes(x = x, y = y))
  # 用withCallingHandlers包裹调用geom_line的代码，捕获警告
  plot <- plot + withCallingHandlers(
    do.call("geom_line", c(list(...))),
    warning = function(w) {
      # 匹配geom_line抛出的未知参数特定警告
      if (grepl("^Ignoring unknown parameters: ", w$message)) {
        # 把警告内容存入外部变量
        caught_warn <<- w$message
        # 屏蔽原始警告输出
        invokeRestart("muffleWarning")
      }
    }
  )
  
  # 如果捕获到了目标警告，自定义修改后重新抛出
  if (!is.null(caught_warn)) {
    # 追加你需要的stat参数，也可以替换成其他你需要新增的参数名
    new_warn <- paste0(caught_warn, ", stat")
    warning(new_warn, call. = FALSE)
  }
  
  return(plot)
}

# 测试代码
data <- tibble::tibble(x = c(1,2,3,4,5,6), y = c(5,6,7,8,9,10))
draw_plot(data, test = 2, wrong = "")

效果说明 #

调用上述测试代码时，会输出你需要的自定义警告：

Warning message:
Ignoring unknown parameters: test, wrong, stat 

核心逻辑说明 #

• withCallingHandlers是基础R提供的条件处理函数，不会中断代码执行，适合捕获警告类非致命异常
• 用正则匹配筛选geom_line抛出的特定未知参数警告，避免误捕获其他不相关警告
• invokeRestart("muffleWarning")用来屏蔽原始的默认警告输出，只保留你自定义的警告
• 捕获到的原始警告存储在caught_warn变量中，你可以按需修改内容后再通过warning()抛出新警告

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