如何让R语言函数支持可变数量的输入参数？

首先修正原函数的语法错误：filter()里的相等判断必须用==而非=，否则会触发赋值逻辑导致报错。

方案1：给参数设置默认NULL值 #

给非必填参数设置默认值NULL，在函数内部动态判断并构建过滤条件，兼容1-3个参数的传入场景：

library(dplyr)

my_fct <- function(input1, input2 = NULL, input3 = NULL) {
  # 初始化基础过滤条件
  condition <- quo(x == input1)
  
  # 若传入input2，追加或条件
  if (!is.null(input2)) {
    condition <- quo(!!condition | x == input2)
  }
  
  # 若传入input3，追加或条件
  if (!is.null(input3)) {
    condition <- quo(!!condition | x == input3)
  }
  
  # 执行过滤并返回结果
  output <- df %>% filter(!!condition)
  return(output)
}

调用示例：

• 仅传1个参数：my_fct(3)
• 传2个参数：my_fct(3, 5)
• 传3个参数：my_fct(3, 5, 7)

方案2：使用...接收任意数量参数（更简洁） #

如果需求是将所有传入的参数值作为x的匹配项，用...可以接收任意数量的输入，合并后用%in%过滤，扩展性更强：

library(dplyr)

my_fct <- function(...) {
  # 合并所有传入的参数为一个向量
  match_values <- c(...)
  # 过滤x属于匹配值的行
  output <- df %>% filter(x %in% match_values)
  return(output)
}

调用示例：

• 单个值：my_fct(3)
• 多个离散值：my_fct(3, 5, 7)
• 混合向量与离散值：my_fct(c(1:10), 15)

额外优化建议 #

避免依赖全局变量df，建议将其作为函数参数传入，让函数更健壮：

my_fct <- function(df, ...) {
  match_values <- c(...)
  df %>% filter(x %in% match_values)
}

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