如何按物种拆分矩阵列表并生成对应动物路径热力图？

1. 重构导入逻辑：绑定物种元数据而非修改矩阵结构 #

核心思路是不改动720x720矩阵本身，而是给每个矩阵绑定对应的物种ID元数据，用嵌套列表的方式存储，既保留矩阵的二维结构，又能关联物种信息。

示例代码：

# 目标文件夹路径（替换为你的实际路径）
matrix_dir <- "your_matrix_folder"

# 获取所有矩阵文件（根据实际文件格式调整后缀，比如.txt/.csv/.RData）
file_paths <- list.files(matrix_dir, full.names = TRUE, pattern = "\\.(txt|csv)$")

# 循环导入并绑定物种ID
matrix_with_meta <- lapply(file_paths, function(path) {
  # 从文件名提取speciesID（根据你的文件名格式调整正则，示例格式："speciesA_sample01.txt"）
  file_name <- basename(path)
  species_id <- sub("_.*\\.(txt|csv)$", "", file_name)
  
  # 导入矩阵（如果是RData格式，替换为 mat <- get(load(path)) ）
  mat <- read.table(path, header = FALSE, sep = " ")
  
  # 返回包含矩阵和物种ID的子列表
  list(matrix = mat, species = species_id)
})

2. 按物种分组 #

用基础R或dplyr工具将列表按物种ID拆分，得到每个物种对应的矩阵集合：

# 方法1：基础R实现
species_groups <- split(matrix_with_meta, sapply(matrix_with_meta, function(x) x$species))

# 方法2：dplyr+purrr实现（更直观）
library(dplyr)
library(purrr)

species_groups <- matrix_with_meta %>%
  map_dfr(~ tibble(species = .x$species, matrix = list(.x$matrix))) %>%
  group_by(species) %>%
  group_split()

3. 批量生成物种专属热力图 #

复用你之前生成多物种热力图的逻辑，循环处理每个物种组即可：

# 以ggplot2可视化为例（替换为你实际用的绘图工具）
library(ggplot2)
library(reshape2)

# 循环每个物种分组生成热力图
for (group in species_groups) {
  # 获取当前物种名
  current_species <- unique(group$species)
  
  # 合并组内所有矩阵（按像素位置累加，和你之前合并多物种的逻辑一致）
  combined_mat <- Reduce("+", group$matrix)
  
  # 转换为长格式适配ggplot
  mat_long <- melt(combined_mat)
  
  # 生成热力图并保存
  p <- ggplot(mat_long, aes(x = Var1, y = Var2, fill = value)) +
    geom_tile() +
    scale_fill_viridis_c(option = "plasma") +
    labs(title = paste("Path Heatmap -", current_species),
         x = "X Pixel", y = "Y Pixel") +
    theme_minimal() +
    theme(plot.title = element_text(hjust = 0.5))
  
  ggsave(paste0(current_species, "_heatmap.png"), p, width = 8, height = 8, dpi = 300)
}

关键注意事项 #

• 绝对不要修改矩阵的二维结构（比如新增列），热力图依赖720x720的像素对应关系，结构改动会直接导致可视化逻辑失效。
• 文件名提取物种ID的正则需根据你的实际命名规则调整，比如如果文件名格式是"sample02_speciesB.csv"，正则要改为sub("^.*_(.*)\\.csv$", "\\1", file_name)。
• 如果矩阵存储为RData格式，导入时需用mat <- get(load(path))正确提取矩阵对象。

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