Python 图文：小白也能轻松生成精美 PDF 报告！

摘要:  还在为枯燥的数据报表发愁吗？想让你的 Python 项目报告瞬间高大上？本文将带你学习如何使用 Python 生成图文并茂的 PDF 文件，从此告别单调，让你的数据“活”起来！

一、 引言

想象一下，你正在为公司做一个用户行为分析，如果只是把一堆表格数据扔给老板，他可能很难理解。但如果你能用 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 文档！