如何修复ggplot2代码，在单个PDF中生成6个匹配标记散点图？

原代码核心问题 #

• 绘图逻辑在for循环外部，仅处理最后一个样本文件的数据，无法批量生成6张图
• ggplot对象在循环中需显式调用print()才能输出到PDF设备
• 颜色映射标签不匹配（aes(color = "matched_file1")与scale_color_manual的"truthlist"未对应）
• 多分支if-else判断冗余，可通过数据预处理简化

修复后完整代码 #

# 加载依赖包（若未安装先运行 install.packages("dplyr")）
library(dplyr)
library(ggplot2)

# 读取参考文件
file1 <- read.table("truthlist.tab", header = TRUE)
file2 <- read.table("background.tab", header = TRUE)

# 定义样本文件列表
file_list <- list(
  "all" = "all/Sample101.summary.tab",
  "all-Sort" = "all-Sort/Sample101.summary.tab",
  "right" = "right/Sample101.summary.tab",
  "right-Sort" = "right-Sort/Sample101.summary.tab",
  "left" = "left/Sample101.summary.tab",
  "left-Sort" = "left-Sort/Sample101.summary.tab"
)

# 打开PDF设备（可按需调整尺寸）
pdf("Sample101_plots.pdf", width = 10, height = 6)

# 循环处理每个样本文件
for (i in seq_along(file_list)) {
  sample_name <- names(file_list)[i]
  file_path <- file_list[[i]]
  
  # 读取样本数据
  sample_data <- read.table(paste0("Desktop\\Sample101\\", file_path), header = TRUE)
  
  # 预处理：给每个点标记颜色分组
  sample_data$color_group <- case_when(
    sample_data$Position %in% file1$Position ~ "truthlist",
    sample_data$Position %in% file2$Position ~ "background",
    TRUE ~ "grey"
  )
  
  # 生成绘图
  p <- ggplot(sample_data, aes(x = Position, y = AF)) +
    geom_point(aes(color = color_group), size = 3) +
    scale_color_manual(
      values = c("truthlist" = "red", "background" = "green", "grey" = "grey"),
      breaks = c("truthlist", "background"), # 可选：隐藏灰色点的图例
      labels = c("匹配truthlist", "匹配background") # 可选：自定义图例标签
    ) +
    labs(title = sample_name, x = "Position", y = "AF") +
    theme_bw() # 可选：美化绘图主题
  
  # 输出到PDF（必须显式调用print）
  print(p)
}

# 关闭PDF设备
dev.off()

关键修改说明 #

1. 循环逻辑调整：将数据读取、预处理、绘图全流程移入for循环，确保每个样本都被独立处理
2. 颜色分组预处理：用case_when直接标记每个点的匹配类型，彻底替代冗余的多分支判断
3. 显式输出ggplot：在循环中调用print(p)，确保ggplot对象被渲染并写入PDF文件
4. 颜色映射修正：统一color_group取值与scale_color_manual的键值对应，保证图例显示正确
5. 优化命名：将列表命名改为更直观的样本分组名，便于识别每张图对应的样本

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