如何在R中创建函数，按传入向量指定范围读取多个CSV文件？

嘿，我帮你优化了这个读取CSV的函数，让它更灵活、更符合R的使用习惯，同时完美支持向量参数指定读取范围~

先说说原代码可以优化的地方 #

• 硬编码setwd("my_data")会改变全局工作目录，容易影响其他代码的运行，不如直接指定文件路径更安全
• 用assign把数据框丢到全局环境里，后续管理起来很麻烦，返回一个命名列表会更方便你统一操作这些数据
• 没有对输入的索引向量做合法性检查，如果不小心传入了超出文件总数的索引，会直接报错，加个判断能让函数更健壮

改进后的函数代码 #

read_selected_csvs <- function(file_indices, data_dir = "my_data") {
  # 获取指定目录下所有CSV文件的完整路径和文件名
  csv_full_paths <- list.files(path = data_dir, pattern = "\\.csv$", full.names = TRUE)
  csv_file_names <- list.files(path = data_dir, pattern = "\\.csv$")
  
  # 检查输入的索引是否在有效范围内
  if (length(csv_full_paths) == 0) {
    stop("指定目录下没有找到CSV文件，请检查路径！")
  }
  if (any(file_indices < 1 | file_indices > length(csv_full_paths))) {
    stop("输入的索引超出文件数量范围，当前目录下共有", length(csv_full_paths), "个CSV文件")
  }
  
  # 读取指定索引的CSV文件，返回命名列表
  selected_datasets <- lapply(csv_full_paths[file_indices], read.csv)
  names(selected_datasets) <- csv_file_names[file_indices]
  
  return(selected_datasets)
}

函数使用示例 #

• 如果你想读取第1、3、5个CSV文件，直接传入向量c(1,3,5)就行：

# 读取指定索引的文件
my_data <- read_selected_csvs(c(1,3,5))

# 查看第一个文件的内容
head(my_data[[1]])

# 或者直接用文件名访问（更直观）
head(my_data$"sales_data_2023.csv")

• 如果你要读取的是连续范围，比如第2到第4个文件，用2:4这种向量就行：

my_data_range <- read_selected_csvs(2:4)

这个改进版的优势 #

1. 不污染全局环境：不会修改全局工作目录，也不会在全局环境里生成一堆零散的数据框
2. 灵活性更高：可以通过data_dir参数指定任意存放CSV的目录，不用硬编码路径
3. 健壮性更好：会自动检查目录下有没有CSV文件，以及输入的索引是否有效，报错信息更清晰
4. 便于管理：返回的命名列表可以让你很方便地通过索引或文件名访问对应的数据集

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