R语言循环后如何恢复初始工作目录？求更优解决方案

嘿，我懂你现在遇到的问题——每次调用这个函数后，工作目录就卡在子文件夹里，手动切回去又麻烦，确实头疼！我给你两个解决方案，其中一个是更优雅的最优解，另一个是针对必须切换目录的场景。

最优方案：不修改全局工作目录（强烈推荐） #

其实最稳妥的办法是完全不改变全局的工作目录，因为在函数里修改全局状态（比如工作目录）是不太好的编程习惯，很容易引发这类意外问题。我们可以直接构造每个文件的完整路径来读取，根本不用切换目录。

另外，你的原代码还有一个小bug：当id是多个值的时候（比如id=1:10），它只会处理最后一个id，因为没有遍历所有传入的id值。我一起帮你修复这个问题：

pollutantmean <- function(directory, pollutant, id = 1:332) {
  # 构造目标文件夹的完整路径，全程不用切换目录
  dir_path <- file.path(getwd(), directory)
  
  data <- data.frame()
  
  # 遍历每个传入的id，处理所有文件
  for (i in id) {
    # 用sprintf自动生成带前导零的文件名，比if-else简洁太多
    filename <- sprintf("%03d.csv", i)
    # 构造单个文件的完整路径
    file_path <- file.path(dir_path, filename)
    # 读取文件
    dt <- read.csv(file_path)
    # 合并到总数据框
    data <- rbind(data, dt)
  }
  
  # 计算指定污染物的均值（忽略NA）
  mean(data[, pollutant], na.rm = TRUE)
}

代码说明： #

• 用sprintf("%03d.csv", i)可以自动生成符合要求的文件名：比如i=23会变成023.csv，i=100变成100.csv，省去了繁琐的条件判断。
• 全程使用file.path构造完整路径，不需要调用setwd，所以全局工作目录完全不会被改变。
• 新增了循环遍历所有id值，现在函数可以处理单个或多个id的情况了。

测试一下：调用pollutantmean("specdata", "nitrate", 23)后，再运行getwd()，你会发现工作目录还是原来的初始目录！

备选方案：用on.exit()确保恢复目录 #

如果因为某些原因你必须要切换目录（比如老代码依赖目录切换），那可以用on.exit()函数——它会在函数结束时（不管是正常执行完还是中途出错）自动执行指定的代码，确保恢复原来的工作目录：

pollutantmean <- function(directory, pollutant, id = 1:332) {
  # 保存初始工作目录
  original_wd <- getwd()
  # 注册函数结束时必须执行的操作：切回初始目录
  on.exit(setwd(original_wd))
  
  # 切换到目标文件夹
  target_wd <- file.path(original_wd, directory)
  setwd(target_wd)
  
  data <- data.frame()
  
  for (i in id) {
    filename <- sprintf("%03d.csv", i)
    dt <- read.csv(filename)
    data <- rbind(data, dt)
  }
  
  mean(data[, pollutant], na.rm = TRUE)
}

这个方法的好处是：不管函数运行过程中有没有报错，on.exit()里的代码都会执行，绝对不会让工作目录停留在子文件夹里，比你手动在循环末尾加setwd可靠多了（手动加的话，如果函数中途出错，那行代码就不会执行，目录就会卡住）。

你的原代码问题分析 #

你提到手动在循环末尾加了setwd但没生效，主要有两个原因：

1. 循环逻辑错误：你的原代码没有遍历id向量，只是处理了单个id值，setwd可能只在某个分支里执行，或者函数结束前根本没走到那行代码。
2. 位置不对：即使你加了setwd，也应该放在函数的最后（循环外面），而不是循环里——循环里加的话，每次循环都会切回初始目录，下一次读取文件又要切回去，反而会出错。

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