如何在R中批量生成所有agent_id的ggplot图表并输出到Markdown

问题背景 #

我有如下R数据框：

library(ggplot2)
library(dplyr)

dt <- 
  expand.grid(
    month = seq(1, 12),
    agent_id = seq(1, 1000)
  ) %>% 
  mutate(sales = row_number()) 

目前可以手动筛选单个agent_id生成带对应标题的折线散点图：

src <- dt %>% filter(agent_id == 2)
title <- as.character(src$agent_id)

ggplot(src, aes(x = month, y = sales, group = 1)) +
  geom_line() +
  geom_point() +
  labs(title = title)

但不想逐个编写src <- dt %>% filter(agent_id == n)（n从1到1000甚至更大）的方式实现，请问该如何批量生成所有agent_id对应的图表并输出到Markdown中？

解决方案 #

方法1：使用purrr包批量生成并输出图表 #

purrr的walk函数可以遍历每个agent_id，直接在R Markdown中输出所有图表：

library(purrr)

# 获取所有唯一的agent_id
agent_ids <- unique(dt$agent_id)

# 遍历每个agent_id，生成并打印图表
walk(agent_ids, function(id) {
  src <- dt %>% filter(agent_id == id)
  p <- ggplot(src, aes(x = month, y = sales, group = 1)) +
    geom_line() +
    geom_point() +
    labs(title = paste("Agent ID:", id))
  print(p)
})

方法2：基础R循环实现 #

如果不想额外加载purrr，用基础for循环也可以实现：

# 获取所有唯一的agent_id
agent_ids <- unique(dt$agent_id)

# 循环生成并输出每个图表
for (id in agent_ids) {
  src <- dt %>% filter(agent_id == id)
  p <- ggplot(src, aes(x = month, y = sales, group = 1)) +
    geom_line() +
    geom_point() +
    labs(title = paste("Agent ID:", id))
  print(p)
}

可选：同时保存图表为单独文件 #

如果需要将每个图表保存为独立文件，在循环中加入ggsave即可：

for (id in agent_ids) {
  src <- dt %>% filter(agent_id == id)
  p <- ggplot(src, aes(x = month, y = sales, group = 1)) +
    geom_line() +
    geom_point() +
    labs(title = paste("Agent ID:", id))
  # 保存为png文件，文件名包含agent_id
  ggsave(paste0("agent_sales_", id, ".png"), p, width = 6, height = 4)
  # 在Markdown中输出图表
  print(p)
}

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