Python生成PDF|极客教程

Python生成PDF

在日常工作中，我们经常需要将文字、数据、图表等内容以PDF形式保存或分享。Python中有许多第三方库可以帮助我们生成PDF文件，比如reportlab、fpdf、PyPDF2等。本文将介绍如何使用reportlab库生成PDF文件，并演示一些基本的操作。

安装reportlab库

首先，我们需要安装reportlab库。在命令行中执行以下命令：

pip install reportlab

安装完成后，我们就可以开始使用reportlab库来生成PDF文件了。

生成简单的PDF文件

下面我们将演示如何生成一个简单的PDF文件，包含一段文字和一个表格。

from reportlab.pdfgen import canvas

# 创建一个PDF文件
c = canvas.Canvas("example.pdf")

# 设置字体和字号
c.setFont("Helvetica", 12)

# 写入文字
c.drawString(100, 700, "Hello, World!")

# 绘制表格
c.grid(range(100, 600, 100), range(600, 100, -100))

# 保存PDF文件
c.save()

上面的代码中，我们首先导入canvas类，然后创建一个PDF文件，并设置字体和字号。接着使用drawString方法写入一段文字，最后使用grid方法绘制一个简单的表格。最后调用save方法保存PDF文件。

运行以上代码后，会在当前目录生成一个名为example.pdf的PDF文件。打开文件可以看到效果。

添加图片和图表

除了文字和表格，我们还可以在PDF文件中添加图片和图表。下面的示例演示了如何在PDF文件中添加一张图片和一个柱状图。

from reportlab.pdfgen import canvas
from reportlab.lib.units import inch
from reportlab.graphics.charts.barcharts import VerticalBarChart
from reportlab.graphics.shapes import Drawing

# 创建一个PDF文件
c = canvas.Canvas("example2.pdf")

# 添加图片
c.drawImage("example_image.jpg", 100, 500, width=2*inch, height=2*inch)

# 创建一个柱状图
data = [(10, 20, 30, 40),]
drawing = Drawing(400, 200)
bc = VerticalBarChart()
bc.x = 50
bc.y = 50
bc.height = 125
bc.width = 300
bc.data = data
drawing.add(bc)

# 将图表添加到PDF文件中
drawing.saveState()
drawing.drawOn(c, 100, 300)
drawing.restoreState()

# 保存PDF文件
c.save()

上面的代码中，我们首先使用drawImage方法在PDF文件中添加一张图片。然后创建一个柱状图，并将其添加到PDF文件中。最后保存PDF文件。

运行以上代码后，会在当前目录生成一个名为example2.pdf的PDF文件。打开文件可以看到添加的图片和柱状图。

添加页眉和页脚

有时候我们需要在PDF文件中添加页眉和页脚，以便更好地组织内容。下面的示例演示了如何在PDF文件中添加页眉和页脚。

from reportlab.pdfgen import canvas

# 创建一个PDF文件
c = canvas.Canvas("example3.pdf")

# 添加页眉
c.setFont("Helvetica", 10)
c.drawString(100, 800, "Page Header")

# 添加页脚
c.setFont("Helvetica", 10)
c.drawString(100, 50, "Page Footer")

# 保存PDF文件
c.save()