R语言for循环批量创建同名目录 分类存储对应热图文件

你原有代码存在3个核心问题导致无法实现按名单独存图：

• PlotHeatMap被直接定义为静态Heatmap对象，不是支持传参的函数，循环调用会直接报错
• 文件夹创建逻辑顺序颠倒，且未将输出路径指向新建文件夹，即使建了文件夹图还是存在根目录
• 缺失绘图设备开关逻辑，ComplexHeatmap对象无法直接自动导出为PNG文件

修正后完整可运行代码 #

首次运行需先安装依赖包：

install.packages(c("tidyverse", "data.table", "ComplexHeatmap"))

正式运行代码：

# 加载依赖包
library(tidyverse)
library(data.table)
library(ComplexHeatmap)

# 矩阵生成函数（保留原有逻辑，补全空值兼容）
GetMatrix=function(data,spToSum) {
  newdat<-filter(data,common_name %in% spToSum)
  # 无对应物种数据时返回NULL避免报错
  if(nrow(newdat) == 0) return(NULL)
  matrixdat<-dcast(newdat, site ~ year_season, value.var="num", fun.aggregate = sum)
  matrixdat$site <- as.character(matrixdat$site)
  matrixdat <- matrixdat[order(nchar(matrixdat$site), matrixdat$site)]
  matrixdat <- matrixdat %>% remove_rownames %>% column_to_rownames(var="site")
  as.matrix(matrixdat)
}

# 修正为可传参的热图绘制+导出函数
PlotHeatMap <- function(data, sp_name, fileName, doCluster = FALSE){
  # 空数据直接跳过
  if(is.null(data)) return(message(paste0(sp_name, "无有效数据，跳过")))
  # 生成热图对象
  hp <- Heatmap(round(log(data+1), digits = 1),
                cluster_rows = doCluster,
                cluster_columns = doCluster,
                na_col = "black",
                column_title = "Year/Season",
                row_title = "Site",
                heatmap_legend_param = list(title = "ln(x+1)"))
  # 导出PNG
  png(filename = fileName, width = 800, height = 600, res = 100)
  draw(hp)
  dev.off()
}

# 初始化参数
dataList<-list()
# 替换为你实际的工作目录
setwd('C:/你的实际工作目录路径')
# 测试时可直接用你提供的dput示例数据给dat赋值，splist会自动提取物种名
# 正式运行替换为你自己的splist对象即可
splist <- unique(dat$common_name)
doClusterVal<-FALSE

# 循环生成热图
for(run in 1:length(splist)) {
  # 当前处理的物种名
  sp_name <- splist[run]
  # *第一步先创建对应该物种名的文件夹，已存在也不报错中断
  dir.create(file.path(sp_name), recursive = TRUE, showWarnings = FALSE)
  # 拼接完整保存路径：物种名同名文件夹/对应热图PNG文件
  save_path <- file.path(sp_name, paste0(run, sp_name," - NotClustered.png"))
  # 生成对应统计矩阵
  dataList[[run]]<-GetMatrix(filter(dat,CYR>=2005), sp_name)
  # 绘图并保存到指定路径
  PlotHeatMap(data=dataList[[run]],
              sp_name = sp_name,
              fileName = save_path,
              doCluster=doClusterVal)
  print(paste(run, sp_name, "处理完成"))
}

关键修改点说明 #

• 调整循环执行顺序：先创建同名文件夹，再将图片保存路径指向该文件夹，完全匹配"每个热图存入同名文件夹"的需求
• 将原本写死的热图定义改为可传参的正式函数，支持循环批量调用，补全空数据兼容逻辑，不会因为个别物种无符合条件的数据中断整个循环
• 路径全部用file.path()拼接，自动适配Windows/macOS/Linux的路径分隔符，不会出现斜杠不兼容的路径错误
• 补全PNG导出的设备开关逻辑，设置了通用的导出尺寸和分辨率，可根据自身需求调整width/height/res参数
• 文件夹创建加了showWarnings = FALSE参数，重复运行代码时如果文件夹已经存在不会弹出冗余警告

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