Question

我正在尝试编写一个自定义函数，该函数使用chisq.test（下面是它的一个玩具版本）执行拟合优度测试。我希望该函数具有鲁棒性，因此我使用tryCatch来确保如果指定了无效的概率向量，该函数将返回一个包含NaN s的数据帧。

这是功能-

set.seed(123)

# custom function
custom_prop <- function(var, ratio) {
  tryCatch(
    expr = broom::tidy(stats::chisq.test(
      x = table(var),
      p = ratio
    )),
    error = function(x) {
      tibble::tribble(
        ~statistic, ~p.value, ~parameter,
        NaN, NaN, NaN
      )
    }
  )
}

尝试有效的比率（向量总和为1；按预期工作）

custom_prop(mtcars$am, c(0.6, 0.4))
#> # A tibble: 1 x 4
#>   statistic p.value parameter method                                  
#>       <dbl>   <dbl>     <dbl> <chr>                                   
#> 1   0.00521   0.942         1 Chi-squared test for given probabilities

custom_prop(mtcars$am, c(0.7, 0.3))
#> # A tibble: 1 x 4
#>   statistic p.value parameter method                                  
#>       <dbl>   <dbl>     <dbl> <chr>                                   
#> 1      1.72   0.190         1 Chi-squared test for given probabilities

尝试无效比率（向量的总和不等于1；按预期工作）

custom_prop(mtcars$am, c(0.6, 0.6))
#> # A tibble: 1 x 3
#>   statistic p.value parameter
#>       <dbl>   <dbl>     <dbl>
#> 1       NaN     NaN       NaN

custom_prop(mtcars$am, c(0.7, 0.5))
#> # A tibble: 1 x 3
#>   statistic p.value parameter
#>       <dbl>   <dbl>     <dbl>
#> 1       NaN     NaN       NaN

但是这种方法的问题是用户不知道为什么函数没有返回结果。如果我不使用tryCatch，他们将看到原因-

broom::tidy(stats::chisq.test(
  x = table(mtcars$am),
  p = c(0.7,0.5)
))
#> Error in stats::chisq.test(x = table(mtcars$am), p = c(0.7, 0.5)): probabilities must sum to 1.

我也尝试了提到的here解决方案，但这只会返回NULL而不会显示错误消息-

# custom function
custom_prop2 <- function(var, ratio) {
  tryCatch(
    expr = broom::tidy(stats::chisq.test(
      x = table(var),
      p = ratio
    )),
    error = function(e) {}
  )
}

# trying out invalid ratios
custom_prop2(mtcars$am, c(0.6, 0.6))
#> NULL

所以，我的问题是-
表达式无法正常工作时，有什么方法可以使用tryCatch和打印错误消息？

Answer 1

只需使用错误消息来构建并抛出warning，就可以了：

custom_prop <- function(var, ratio) {
  tryCatch(
    expr = broom::tidy(stats::chisq.test(
      x = table(var),
      p = ratio
    )),
    error = function(x) {
      warning(x)   # just add this line
      tibble::tribble(
        ~statistic, ~p.value, ~parameter,
        NaN, NaN, NaN
      )
    }
  )
}

custom_prop(mtcars$am, c(0.6, 0.6))
# A tibble: 1 x 3
#  statistic p.value parameter
#      <dbl>   <dbl>     <dbl>
#1       NaN     NaN       NaN
#Warning message:
#In stats::chisq.test(x = table(var), p = ratio) :
#  probabilities must sum to 1.

使用`tryCatch`并在表达式失败时打印错误消息

1 个答案: