主页 / coding / 问题 / 79464245
Accepted
robertspierre
Asked: 2025-02-25 00:29:37 +0800 CST

将特定警告类型转换为错误

  • 772

考虑以下代码：

> df <- tibble(gender=c(1,1,0))
> df$male
Warning: Unknown or uninitialised column: `male`.
NULL

如何将这种特定的警告类型转换为错误？

我想要类似的东西options(warn = 2)，但仅限于这种特定类型的警告（即引用不存在的 tibble 中的列）

5 个回答

  1. 这有点儿不靠谱，但你可以修补该函数，即 getter。当在 中找不到列名时，$会抛出警告。通过用 替换此调用，我们可以将此警告转换为错误。rlang::warndfrlang::abort

    patch_tibble <- function() {
  if(is.null(attr(tibble:::`$.tbl_df`, "patched"))) {
    tt <- get("$.tbl_df", envir=asNamespace("tibble"), inherits=FALSE)
    
    body(tt) <- methods::substituteDirect(body(tt), list(warn=quote(abort)))
    attr(tt, "patched") <- TRUE
    
    unlockBinding("$.tbl_df", asNamespace("tibble"))
    assign("$.tbl_df", tt, envir=asNamespace("tibble"), inherits = FALSE)
    lockBinding("$.tbl_df", asNamespace("tibble"))
    packageStartupMessage("Patched tibble!")
  }
}

    library(tibble)
patch_tibble()

df <- tibble(gender=c(1,1,0))
df$male

Error in `df$male`:
! Unknown or uninitialised column: `male`.
Run `rlang::last_trace()` to see where the error occurred.
    • 4
  2. Best Answer

    S3 方法的原始函数$.tbl_df定义如下：

    function (x, name) 
{
    out <- .subset2(x, name)
    if (is.null(out)) {
        warn(paste0("Unknown or uninitialised column: ", tick(name), 
            "."))
    }
    out
}

    stop您可以简单地使用或将警告替换为错误rlang:abort()：

    `$.tbl_df` <- function (x, name) {
  out <- .subset2(x, name)
  if (is.null(out)) {
    rlang::abort(paste0("Unknown or uninitialised column: ", name, "."))
  }
  out
}

    然后，正如预期的那样：

    > df$male
Error in `$.tbl_df`(df, male) : Unknown or uninitialised column: male.

    附录

    正如评论中指出的那样，如果$.tbl_df在内部调用此解决方案，则可能会导致问题。按照@Billy34的提示，我们可以$.tbl_df用修改后的函数替换内部函数，如下所示：

    patch_tibble <- function(){
  
  # The modified "$.tbl_df" function
  f <- function (x, name) {
    out <- .subset2(x, name)
    if (is.null(out)) {
      abort(paste0("Unknown or uninitialised column: ", tick(name), 
                   "."))
    }
    out
  }
  
  # Set the function's namespace to tibble
  environment(f) <- asNamespace("tibble")
  
  # Replace the original function with our modified version
  utils::assignInNamespace("$.tbl_df", f, "tibble")
  
  print("Patched tibble!")
}

    输出结果如下：

    > patch_tibble()
[1] "Patched tibble!"
> df$male
Error in `df$male`:
! Unknown or uninitialised column: `male`.
    • 3

  3. 根据@BenBolker 的回答中的想法，您可以创建一个全局处理程序：

    globalCallingHandlers(warning = function(w) {
  if(grepl("Unknown or uninitialised column", w$message, fixed = TRUE)) 
    stop(errorCondition(w$message, class = "unknown_column_extract_error")) else w
  })


library(tibble)
df <- tibble(gender=c(1,1,0))
df$male
#Error: Unknown or uninitialised column: `male`.

#doesn't affect other warnings
as.numeric("a")
#[1] NA
#Warning message:
#  NAs introduced by coercion

    当然，如果警告被分类的话这会更简单。

    • 3

  4. 这可能是您能得到的最接近的答案（没有像上一个答案那样进行黑客攻击）：使用tryCatch()检查警告消息（如果有人使用已翻译警告的不同语言设置，此操作将会失败）。

    my_try <- function(expr, warnstr = "Unknown or uninitialised column") { 
   tryCatch(expr,
      warning = function(e) { 
         if(grepl(warnstr, e)) simpleError(e) })
}
my_try(df$male)
    • 2

  5. 其他人已经给出了解决此问题的方法。在此回答中，我将解释tibble作者可以采取哪些措施来简化此工作。

    R 支持分类错误和警告，但它们并不常用。rlang包中对class、 和 都有参数abort()。warn()您inform()不需要使用rlang，所有操作都可以用基本函数完成，但它们很好地打包了它。

    因此，tibble可以做的是，为他们发出的每个警告设置一个单独的类。然后，您可以捕获与您感兴趣的警告相对应的特定类，并将其转换为错误。或者，您可以创建自己的$.tbl_df方法来执行此操作，然后选择是否要进行转换。

    例如，

    warn(paste0("Unknown or uninitialised column: ", tick(name), 
        "."))

    他们本可以写

    warn(paste0("Unknown or uninitialised column: ", tick(name), 
        "."), class = "warn.$.tbl_df")

    然后你可以运行

    tryCatch(df$male, `warn.$.tbl_df` = function(w) abort(w$message, class = "error.$.tbl_df", call = NULL))

    捕获该警告，并将其转换为错误。我将其转换为带类的错误，"error.$.tbl_df"以便其他代码可以捕获该错误，但没有必要这样做。我还设置了，call = NULL这样您就不会被告知代码中的一行$.tbl_df对您毫无意义。如果您尝试设置，您也许能够找到df$male表达式；我不确定它会回溯到多远。

    • 2

相关问题