如何实现重复提示用户下载Python绘图及批量保存功能？

1. 封装单张图表保存的交互逻辑 #

把重复的输入判断和保存代码封装成函数，每次生成图表后直接调用即可，彻底避免重复编写相同逻辑：

import matplotlib.pyplot as plt

def save_figure(fig_name, dpi=100):
    save_choice = input(f"Would you like to save this figure as {fig_name}?")
    yes_options = {'yes', 'Yes', 'YES', 'y', 'Y'}  # 多支持几种简短输入更友好
    if save_choice in yes_options:
        plt.savefig(f"{fig_name}.png", dpi=dpi, bbox_inches='tight')
        print(f"已保存为 {fig_name}.png")
    else:
        print("Ok :(")

使用示例：

# 生成示例折线图
plt.plot([1,2,3], [4,5,6])
plt.title("Sample Line Plot")
# 直接调用封装好的保存函数
save_figure("line_plot")

2. 实现批量保存所有图表的功能 #

要实现批量保存，需要在生成每张图表时记录对应的Figure对象，最后统一处理：

# 初始化列表存储所有生成的图表对象
all_figures = []

# 生成第一张图表
plt.plot([1,3,5], [2,4,6])
plt.title("Plot 1")
all_figures.append(plt.gcf())  # 获取当前图表对象并加入列表
save_figure("plot_1")  # 执行单张保存交互
plt.close()  # 关闭当前图表，释放内存并避免后续图表重叠

# 生成第二张图表
plt.bar([1,2,3], [10,20,15])
plt.title("Plot 2")
all_figures.append(plt.gcf())
save_figure("plot_2")
plt.close()

# 程序结束后提供批量保存选项
batch_save = input("是否要批量保存所有图表？")
if batch_save in {'yes', 'Yes', 'YES', 'y', 'Y'}:
    for idx, fig in enumerate(all_figures, 1):
        fig.savefig(f"batch_plot_{idx}.png", dpi=100, bbox_inches='tight')
        print(f"已批量保存 batch_plot_{idx}.png")
else:
    print("批量保存已取消")

补充细节 #

• 用集合yes_options代替元组，查找效率更高，同时支持简短输入（如'y'）提升用户体验
• 生成图表后调用plt.close()，在Jupyter Notebook中能避免图表重叠显示，还能减少内存占用
• 批量保存时用enumerate自动生成序号，无需手动给每张批量保存的图表命名

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