R自定义函数中如何实现数据logit变换的条件触发逻辑？

前置修复：补全自定义logit函数 #

原自定义logit函数缺少返回值，补全同时增加输入校验避免非法值报错：

logit <- function(x){
  # 校验输入值范围，避免0/1外的数值导致计算异常
  if(any(x < 0 | x > 1, na.rm = TRUE)){
    warning("logit变换要求输入值位于0-1区间，异常值将返回NA")
  }
  logprop <- log(x / (1 - x))
  return(logprop)
}

修正后的plot_cp函数 #

plot_cp <- function(est, iso_code, logit_scale = FALSE){
  # 第一步：完成数据筛选
  newdata <- fp.data %>% 
    filter(`ISO code` == iso_code, 
           `Median estimate and uncertainty intervals` == "Median", 
           Indicator == "CPmod", 
           `Marital status` == "MW")
  
  # 第二步：按参数条件执行logit变换
  if(logit_scale == TRUE){
    # 注意：如果你的Percentage列是0-100的百分比值，需要先除以100转成0-1比例再计算，即新增一行 newdata$Percentage <- newdata$Percentage / 100
    # 做logit变换同时截断下限为logit(0.01)，避免极端值返回Inf
    newdata$Percentage <- pmax(logit(0.01), logit(newdata$Percentage))
  }
  
  # 第三步：生成绘图结果
  p <- ggplot(data = newdata) + 
    geom_line(mapping = aes(Year, Percentage)) +
    # 可选优化：y轴标签随变换自动调整
    labs(y = ifelse(logit_scale, "Logit(Contraceptive Prevalence)", "Contraceptive Prevalence (%)"))
  
  return(p)
}

核心修改说明 #

• 移除了原代码中对logit_scale参数的错误重定义：原代码将布尔类型的入参logit_scale覆盖为了一个自定义函数，导致传入的参数完全失效
• 新增标准if条件判断分支，仅当传入logit_scale = TRUE时才执行logit变换逻辑
• 补全了logit函数的返回值与输入校验，避免非法输入导致程序崩溃
• 可选增加了y轴标签自动适配逻辑，提升绘图可读性

小提示：当前函数定义的入参est未在函数内使用，可根据实际需求补充对应逻辑，或移除冗余参数。

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