R语言如何统计自定义函数生成的plot图表的数量

下面分两种适用场景给出可直接运行的方案，适配你提到的ggplot2绘图场景：

方案1：可修改原绘图函数的场景 #

直接在函数内部嵌入计数逻辑即可，逻辑简单性能损耗极低，示例代码如下：

# 修改后的绘图函数，最终返回生成的图表数量
ploting = function(x) {
  plot_count <- 0
  for(i in 1:x){
    # 这里替换为你自己的ggplot2打印语句即可，比如print(ggplot(...) + ...)
    print(plot(rnorm(10)))
    plot_count <- plot_count + 1
  }
  return(plot_count)
}

# 调用函数直接获取计数
result_count <- ploting(5)
print(result_count) # 示例输出为5

方案2：无需修改原函数的场景 #

可以利用R的虚拟图形设备实现无侵入统计，原理是把所有绘图输出到临时PDF文件，通过统计PDF页数得到图表总数，不会弹出多余绘图窗口：

# 原函数无需任何修改
ploting = function(x)
{
  for(i in 1:x){
    print(plot(rnorm(10)))
  }
}

# 生成临时PDF文件路径
temp_pdf <- tempfile(fileext = ".pdf")
# 开启PDF虚拟设备，所有后续绘图都会写入该文件
pdf(temp_pdf)

# 运行你的绘图函数
ploting(5)

# 关闭虚拟设备
dev.off()

# 统计PDF页数即为图表数量，用R自带函数无需额外安装包
plot_count <- grDevices::pdf.info(temp_pdf)$Pages
print(plot_count) # 示例输出为5

# 可选：删除临时文件释放空间
unlink(temp_pdf)

注意：如果你的ggplot2绘图代码没有显式print()调用，在函数内部运行时不会输出到图形设备，两种方案都会统计不到，确保你的绘图逻辑在函数内执行时会触发图形输出即可。

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