如何让R语言函数同时支持文件路径与环境对象作为输入参数？

方案一：利用R原生S3方法（最推荐） #

S3方法是R中处理不同输入类型的原生机制，能让函数根据输入对象的类型自动调用对应逻辑，完全避免if-else嵌套，代码更清晰易维护。

实现代码： #

1. 定义泛型函数：

myfunction <- function(x, ...) {
  UseMethod("myfunction")
}

2. 针对文件路径（字符型）的处理方法：

myfunction.character <- function(x, ...) {
  library(vroom)
  library(dplyr)
  
  vroom(x, ...) %>%
    filter(qc_tag == "PASS")
}

3. 针对数据框/ tibble的处理方法：

myfunction.data.frame <- function(x, ...) {
  library(dplyr)
  
  x %>%
    filter(qc_tag == "PASS")
}

调用时无需额外判断，函数会自动适配输入类型：

• 传入文件路径字符串时，自动触发读取+过滤逻辑；
• 传入已有的数据框/tibble时，直接执行过滤操作。

方案二：轻量类型判断方案（无需S3） #

如果不想用S3方法，可借助rlang包的类型判断函数，把逻辑压缩为简洁的分支，比嵌套if-else更清爽：

library(vroom)
library(dplyr)
library(rlang)

myfunction <- function(x, ...) {
  # 统一数据来源：路径则读取，已有表格则直接使用
  data <- if (inherits_any(x, c("character", "factor"))) {
    vroom(x, ...)
  } else {
    x
  }
  
  # 统一执行过滤逻辑
  data %>%
    filter(qc_tag == "PASS")
}

额外优化：添加输入合法性校验 #

可以加入简单的校验，避免非法输入导致的报错：

myfunction <- function(x, ...) {
  if (!inherits_any(x, c("character", "factor", "data.frame", "tbl_df"))) {
    stop("输入必须是文件路径（字符/因子）或数据框/tibble")
  }
  
  data <- if (inherits_any(x, c("character", "factor"))) {
    vroom(x, ...)
  } else {
    x
  }
  
  data %>%
    filter(qc_tag == "PASS")
}

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