如何在Python DataFrame多页PDF生成中避开with代码块

当然可以避开with代码块，核心思路是手动管理PdfPages对象的生命周期，下面是两种可行的方案：

1. 手动创建和关闭PdfPages对象 #

直接实例化PdfPages对象，完成所有图表保存后手动调用close()方法即可，替代with的自动管理逻辑：

from matplotlib.backends.backend_pdf import PdfPages
import matplotlib.pyplot as plt

# 手动创建PdfPages实例
pdf = PdfPages('multipage_pdf.pdf')

# 生成第一张图表
plt.figure(figsize=(3, 3))
plt.plot(range(7), [3, 1, 4, 1, 5, 9, 2], 'r-o')
plt.title('Page One')
pdf.savefig()
plt.close()

# 生成更多图表（示例）
plt.figure(figsize=(3, 3))
plt.bar(range(5), [2, 5, 3, 7, 1])
plt.title('Page Two')
pdf.savefig()
plt.close()

# 必须手动关闭，否则PDF文件会损坏
pdf.close()

注意：一定要记得调用pdf.close()，否则PDF文件无法正常生成或打开——with语句的核心作用就是自动处理对象的关闭（即使发生异常），手动管理时需要自己确保这一步执行。

2. 封装保存逻辑，拆分代码 #

如果代码量很大，可以把图表生成和保存的逻辑封装成函数，进一步简化主代码结构，同时避开大段with包裹：

from matplotlib.backends.backend_pdf import PdfPages
import matplotlib.pyplot as plt

def save_single_page(pdf_obj, title, plot_action):
    """封装单张图表的生成与保存逻辑"""
    plt.figure(figsize=(3, 3))
    plot_action()  # 执行图表绘制逻辑
    plt.title(title)
    pdf_obj.savefig()
    plt.close()

# 手动创建PdfPages实例
pdf = PdfPages('multipage_pdf.pdf')

# 调用函数生成图表
save_single_page(pdf, 'Page One', lambda: plt.plot(range(7), [3, 1, 4, 1, 5, 9, 2], 'r-o'))
save_single_page(pdf, 'Page Two', lambda: plt.bar(range(5), [2, 5, 3, 7, 1]))
# 可以继续添加更多图表...

# 关闭PDF文件
pdf.close()

这种方式把重复的创建、保存、关闭逻辑抽离，主代码只需要关注图表的具体绘制逻辑，即使代码很长也能保持清晰。

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