一、 引言
想象一下，你正在为公司做一个用户行为分析，如果只是把一堆表格数据扔给老板，他可能很难理解。但如果你能用 Python 生成一份精美的 PDF 报告，包含了各种图表、分析结论，甚至还有代码示例，是不是瞬间就高大上了呢？
二、 Python 生成 PDF 的利器：ReportLab 库
Python 社区提供了丰富的库来处理 PDF 文件，其中 ReportLab 库功能强大且易于上手，非常适合用来生成图文并茂的报告。
安装 ReportLab:
pip install reportlab
三、 小试牛刀：生成简单的 PDF 文档
让我们先从一个简单的例子开始，了解 ReportLab 的基本用法：
from reportlab.pdfgen import canvas # 创建一个 PDF 文件 c = canvas.Canvas("hello.pdf") # 设置字体和字号 c.setFont("Helvetica", 16) # 在指定位置写入文本 c.drawString(100, 750, "Hello, world!") # 保存 PDF 文件 c.save()
代码解读:
1、from reportlab.pdfgen import canvas: 导入 canvas 模块，它提供了创建 PDF 文件的基本功能。
2、c = canvas.Canvas("hello.pdf"): 创建一个名为 "hello.pdf" 的 PDF 文件对象。
3、c.setFont("Helvetica", 16): 设置字体为 Helvetica，字号为 16。
4、c.drawString(100, 750, "Hello, world!"): 在坐标 (100, 750) 处写入文本 "Hello, world!"。
5、c.save(): 保存 PDF 文件。
运行这段代码，你会在当前目录下得到一个名为 "hello.pdf" 的文件，打开它，你就会看到 "Hello, world!" 这段文字。
四、 图文并茂：插入图片和表格
除了文本，ReportLab 还支持插入图片、表格等元素，让你的 PDF 报告更加丰富多彩。
插入图片:
from reportlab.lib.units import cm from reportlab.pdfgen import canvas # 创建一个 PDF 文件 c = canvas.Canvas("hello.pdf") # 设置字体和字号 c.setFont("Helvetica", 16) # 在指定位置写入文本 c.drawString(100, 750, "Hello, world!") # 插入图片 c.drawImage("logo.png", 100, 650, width=5*cm, height=3*cm) # 保存 PDF 文件 c.save()
代码解读:
- from reportlab.lib.units import cm: 导入单位转换模块，方便我们使用厘米作为单位。
- c.drawImage("logo.png", 100, 650, width=5*cm, height=3*cm): 在坐标 (100, 650) 处插入名为 "logo.png" 的图片，并设置图片的宽度为 5 厘米，高度为 3 厘米。
请注意: 你需要将一张名为 "logo.png" 的图片放置在与代码相同的目录下，或者修改代码中的图片路径。
插入表格:
from reportlab.lib.units import cm from reportlab.pdfgen import canvas from reportlab.platypus import SimpleDocTemplate, Table, TableStyle from reportlab.lib import colors # 创建一个 PDF 文件 c = canvas.Canvas("hello.pdf") # 设置字体和字号 c.setFont("Helvetica", 16) # 在指定位置写入文本 c.drawString(100, 750, "Hello, world!") # 插入图片 c.drawImage("logo.png", 100, 650, width=5*cm, height=3*cm) # 创建表格数据 data = [['姓名', '年龄', '性别'], ['张三', 25, '男'], ['李四', 30, '女']] # 创建表格对象 table = Table(data) # 设置表格样式 style = TableStyle([ ('BACKGROUND', (0, 0), (-1, 0), colors.grey), ('TEXTCOLOR', (0, 0), (-1, 0), colors.whitesmoke), ('ALIGN', (0, 0), (-1, -1), 'CENTER'), ('FONTNAME', (0, 0), (-1, 0), 'Helvetica-Bold'), ('FONTSIZE', (0, 0), (-1, 0), 14), ('BOTTOMPADDING', (0, 0), (-1, 0), 12), ('BACKGROUND', (0, 1), (-1, -1), colors.beige), ('GRID', (0, 0), (-1, -1), 1, colors.black) ]) table.setStyle(style) # 将表格添加到 PDF 文档 table.wrapOn(c, 100, 500) table.drawOn(c, 100, 500) # 保存 PDF 文件 c.save()
代码解读:
1、from reportlab.platypus import SimpleDocTemplate, Table, TableStyle: 导入 Table 和 TableStyle 模块，用于创建和设置表格样式。
2、data = [['姓名', '年龄', '性别'], ...]: 创建一个二维列表作为表格数据。
3、table = Table(data): 创建一个表格对象。
4、style = TableStyle([...]): 创建一个表格样式对象，设置表格的背景颜色、字体颜色、对齐方式、边框等。
5、table.setStyle(style): 将样式应用到表格。
6、table.wrapOn(c, 100, 500): 计算表格的尺寸，为其在 PDF 文档中预留空间。
7、table.drawOn(c, 100, 500): 在坐标 (100, 500) 处绘制表格。
五、 实际案例：生成股票分析报告
以下是一个简单的股票分析报告的例子，你需要预先生成 K 线图并保存为 "k_line_chart.png" 文件：
from reportlab.pdfgen import canvas from reportlab.lib.units import cm from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Image from reportlab.lib.styles import getSampleStyleSheet # 假设我们已经获取了股票数据和分析结果 stock_name = "阿里巴巴" stock_code = "BABA" analysis_result = "预计未来一年股价将上涨 10%..." # 创建 PDF 文档 doc = SimpleDocTemplate("stock_report.pdf") # 创建故事流，用于存储 PDF 内容 story = [] # 添加标题 styles = getSampleStyleSheet() title = Paragraph(f"股票分析报告：{stock_name} ({stock_code})", styles['Heading1']) story.append(title) story.append(Spacer(1, 2*cm)) # 添加分析结果 paragraph = Paragraph(analysis_result, styles['Normal']) story.append(paragraph) story.append(Spacer(1, 1*cm)) # 添加 K 线图 (假设已经生成图片文件) image = Image("k_line_chart.png", width=15*cm, height=8*cm) story.append(image) # 生成 PDF 文件 doc.build(story)
请注意: 你需要将一张名为 "k_line_chart.png" 的图片放置在与代码相同的目录下，或者修改代码中的图片路径。
六、 总结
本文介绍了使用 Python 生成图文并茂 PDF 文件的基本方法，并通过股票分析报告的例子展示了 ReportLab 库的强大功能。当然，ReportLab 还有更多高级特性等待你去探索，比如自定义页面布局、添加水印、生成目录等等。相信通过不断学习和实践，你一定能用 Python 创建出更加精美实用的 PDF 文档！
