必须用 plt.savefig() 显式保存图像,而非 plt.show();每次循环需新建画布、规范路径、设置 bbox_inches 和 dpi、及时 plt.close();函数应支持动态样式参数和中文显示;文件名须清洗非法字符并加时间戳防重名。

用 plt.savefig() 替代 plt.show() 才能批量存图
很多人写完绘图循环发现文件夹空空如也,根本原因是没关掉交互模式或误用 plt.show()。这个函数只负责弹窗显示,不保存任何文件,且会阻塞后续执行(尤其在无 GUI 环境如服务器上直接报错)。必须显式调用 plt.savefig(),并确保每次调用前图形对象是“新鲜”的。
实操建议:
- 每次循环开头用
plt.figure(figsize=(8, 5))或fig, ax = plt.subplots(figsize=(8, 5))新建画布,避免图像叠加 -
plt.savefig()路径必须是完整文件路径(含后缀),例如"./reports/sales_2024Q1.png",相对路径容易因工作目录变化失效 - 加参数
bbox_inches='tight'防止标题/标签被截断;dpi=150保证业务报表清晰度 - 循环末尾务必加
plt.close()(或plt.close(fig)),否则内存持续增长,跑几百张时大概率 OOM
封装绘图函数时,把数据和配置项当参数传进去
硬编码坐标轴范围、颜色、字体的函数没法复用。真正可批量的关键,是让函数只关心“画什么”,不关心“画成什么样”——样式细节应由调用方按需注入。
常见错误现象:所有图的 y 轴范围一样,但某个月销售额突增导致柱子压顶;或者中文标题全变成方块。
实操建议:
- 函数签名类似:
def plot_sales_report(data, title="", ylabel="万元", ylim=None, save_path=None): - 内部用
ax.set_ylim(ylim),传入ylim=(0, data["value"].max() * 1.1)就自动适配每张图 - 中文字体问题统一在函数开头处理:
plt.rcParams['font.sans-serif'] = ['SimHei', 'Arial Unicode MS'];plt.rcParams['axes.unicode_minus'] = False - 如果
save_path为None,就跳过保存,方便调试时只看图
循环生成文件名时,避免非法字符和重名
直接用业务字段拼接路径,比如 f"{row['region']}_{row['month']}.png",遇到 “华东/华南” 或 “2024-03” 这类值就会出错:斜杠是路径分隔符,短横线在某些系统里触发意外行为,空格导致命令行解析失败。
使用场景:从 pandas DataFrame 的每一行生成一张图,region、product、date 都是变量。
实操建议:
- 用
re.sub(r'[^a-zA-Z0-9_\-.]', '_', text)清洗字符串,把所有非法字符替换成下划线 - 日期统一转为
pd.to_datetime(row['date']).strftime('%Y%m%d'),杜绝格式歧义 - 加时间戳前缀防重名:
f"report_{int(time.time())}_{clean_name}.png",尤其多人同时运行脚本时 - 提前检查目录存在:
os.makedirs(os.path.dirname(save_path), exist_ok=True),别让第一张图就因路径不存在而中断
性能瓶颈常在 I/O 和字体渲染,不是绘图本身
跑 500 张图耗时 3 分钟,其中 2 分半花在磁盘写入和字体加载上。Matplotlib 每次 savefig 都会重新解析字体、计算布局、写入 PNG 编码流——这些操作无法跳过,但可以压缩开销。
参数差异与影响:
-
format='png'是默认,但若接受稍低质量,改用format='jpg'+quality=95可提速 20%~30% - 关闭抗锯齿:
plt.rcParams['path.simplify'] = True和plt.rcParams['path.simplify_threshold'] = 1.0,对柱状图/折线图几乎无视觉损失 - 不要在循环里反复调用
plt.style.use('seaborn'),移到函数外一次性设置 - 如果报表结构高度一致(比如都是同比柱状图),考虑用
matplotlib.animation.FuncAnimation批量渲染帧再导出,但开发成本高,仅适合超大批量(>2000)
真正卡住的时候,先 time.time() 包住 savefig 行,确认是不是磁盘慢;再检查是否漏了 plt.close() 导致后台积压了数百个未释放的 figure 对象。
Python免费学习笔记(深入):立即使用
在学习笔记中,你将探索 Python 的核心概念和高级技巧!











