R语言概率分布函数标量参数取值边界自动检测方案问询

需求说明 #

无需查阅官方文档，自动识别R中概率分布函数的标量参数可接受取值范围，参数类型限定为以下四类：

• 介于-Inf与Inf之间的double类型
• 介于0与Inf之间的double类型
• 正整数
• 介于0与1之间的double类型

注：仅处理含标量参数的分布，排除多项分布这类含向量参数的分布。期望实现类似二项分布的检测结果：size为正整数、prob为0-1之间的double类型。

现有代码问题 #

你提供的代码核心问题在于未生成有效的边界候选值，也没有针对单个参数逐一测试的逻辑，循环与错误捕获的逻辑也不完整，无法区分不同参数的取值限制。

改进后的实现代码 #

checkArgBoundaries <- function(distribution) {
  # 获取分布的随机生成函数
  r_dist <- get(paste0("r", distribution), mode = "function")
  # 获取函数参数，排除n（样本量）和ncp（非中心参数，部分分布有）
  params <- setdiff(names(formals(r_dist)), c("n", "ncp"))
  
  # 针对不同参数类型，生成测试候选值
  candidate_sets <- list(
    all_real = c(-Inf, -100, -1, 0, 1, 100, Inf),
    non_neg_real = c(-1, -0.1, 0, 0.5, 1, 100, Inf),
    prob_range = c(-0.1, 0, 0.5, 1, 1.1),
    positive_integer = c(-2, -1, 0, 0.5, 1, 2, 3)
  )
  
  # 存储每个参数的检测结果
  results <- list()
  
  for (param in params) {
    # 获取参数默认值，作为其他参数的固定值
    default_args <- formals(r_dist)
    default_args[["n"]] <- 1  # 固定生成1个样本
    # 移除当前测试参数的默认值，后续替换为候选值
    default_args[[param]] <- NULL
    
    # 测试所有候选值，记录哪些值会触发错误/警告
    test_results <- lapply(candidate_sets, function(candidates) {
      sapply(candidates, function(val) {
        current_args <- c(default_args, setNames(list(val), param))
        # 捕获错误和警告
        res <- tryCatch({
          do.call(r_dist, current_args)
          TRUE  # 无报错/警告则合法
        }, error = function(e) FALSE,
        warning = function(w) FALSE)
        res
      })
    })
    
    # 根据测试结果判断参数类型
    param_type <- case_when(
      all(test_results$all_real) ~ "介于-Inf与Inf之间的double类型",
      all(test_results$non_neg_real[c(-1, -2)]) & !any(test_results$non_neg_real[c(1,2)]) ~ "介于0与Inf之间的double类型",
      all(test_results$positive_integer[c(5,6,7)]) & !any(test_results$positive_integer[c(1,2,3,4)]) ~ "正整数",
      all(test_results$prob_range[c(2,3,4)]) & !any(test_results$prob_range[c(1,5)]) ~ "介于0与1之间的double类型",
      TRUE ~ "无法识别的参数类型"
    )
    
    results[[param]] <- param_type
  }
  
  return(results)
}

# 测试示例：二项分布
checkArgBoundaries("binom")
# 输出：$size [1] "正整数"；$prob [1] "介于0与1之间的double类型"

# 测试示例：正态分布
checkArgBoundaries("norm")
# 输出：$mean [1] "介于-Inf与Inf之间的double类型"；$sd [1] "介于0与Inf之间的double类型"

代码说明 #

1. 参数提取：通过formals()获取分布函数的参数，排除通用的n（样本量）和ncp（非中心参数）。
2. 候选值设计：针对四类参数类型，分别设计包含边界值、合法值、非法值的候选集合，覆盖可能的取值边界。
3. 逐一测试：对每个参数，固定其他参数为默认值，替换当前参数为候选值，通过tryCatch捕获错误和警告，判断值是否合法。
4. 类型判断：根据候选值的测试结果，匹配对应的参数类型规则，输出最终结论。

内容的提问来源于stack exchange，提问作者Aku-Ville Lehtimäki