PDF自动化最热门的4种Python库！

Python可以实现PDF的自动化文件的自动化处理，比如自动化生成PDF报告、提取PDF中的数据、合并多个PDF文件等。

下面介绍4种常用的的自动化处理PDF的Python库。

PyPDF2库

PyPDF2是Python中常用的处理PDF文件的第三方库，可以用于创建、编辑、合并、拆分、提取、解密等PDF文件操作。

下面是PyPDF2库的一些常见用法：

读取PDF文件

可以使用PdfFileReader类读取PDF文件，示例代码如下：

import PyPDF2

with open('example.pdf', 'rb') as pdf_file:
    pdf_reader = PyPDF2.PdfFileReader(pdf_file)
    num_pages = pdf_reader.getNumPages()
    print(f'Total pages: {num_pages}')

写入PDF文件

使用PdfFileWriter类创建一个PDF文件，并将多个页面添加到该文件中，示例代码如下：


import PyPDF2

pdf_writer = PyPDF2.PdfFileWriter()

for i in range(3):
    pdf_writer.addBlankPage(width=72, height=72)

with open('output.pdf', 'wb') as output_file:
    pdf_writer.write(output_file)

拆分PDF文件

可以使用PdfFileWriter和PdfFileReader类将PDF文件拆分成多个单独的文件，示例代码如下：

import PyPDF2

with open('example.pdf', 'rb') as pdf_file:
    pdf_reader = PyPDF2.PdfFileReader(pdf_file)

    for i in range(pdf_reader.getNumPages()):
        pdf_writer = PyPDF2.PdfFileWriter()
        pdf_writer.addPage(pdf_reader.getPage(i))

        with open(f'page{i+1}.pdf', 'wb') as output_file:
            pdf_writer.write(output_file)

合并PDF文件

可以使用PdfFileMerger类将多个PDF文件合并成一个文件，示例代码如下：


import PyPDF2

pdf_merger = PyPDF2.PdfFileMerger()

pdf_merger.append('file1.pdf')
pdf_merger.append('file2.pdf')
pdf_merger.append('file3.pdf')

with open('merged_file.pdf', 'wb') as output_file:
    pdf_merger.write(output_file)

提取PDF文本

可以使用PdfFileReader类提取PDF文本，示例代码如下：

import PyPDF2

with open('example.pdf', 'rb') as pdf_file:
    pdf_reader = PyPDF2.PdfFileReader(pdf_file)
    page = pdf_reader.getPage(0)
    text = page.extractText()
    print(text)

pdfminer库

pdfminer是一个用于提取PDF文件文本信息的Python库，它可以将PDF文件转换为纯文本或HTML格式，并提供了一些可自定义的参数和配置选项。

pdfminer库包含了pdfminer.six和pdfminer3k两个子模块，分别对应Python 2和Python 3两个版本。它们都提供了相同的功能和API，只是在用法上有所不同。

pdfminer库的主要功能有：

PDF文本提取：可以提取PDF文件中的文本信息，包括文字、字体、字号、坐标等，并按照页面顺序进行排列。
PDF文本转换：可以将PDF文件转换为HTML或XML格式，以便于进行数据分析和处理。
PDF元数据提取：可以提取PDF文件的元数据，包括标题、作者、主题、关键词、创建日期、修改日期等。
PDF页面分析：可以对PDF文件的每一页进行分析，包括获取页面大小、页面旋转角度、页面文本等信息。

我们可以使用pip工具安装pdfminer库，具体命令如下：

pip install pdfminer

使用pdfminer库也非常方便，下面是一个简单的例子，展示了如何提取PDF文件中的文本信息：

from pdfminer.pdfinterp import PDFResourceManager, PDFPageInterpreter
from pdfminer.converter import TextConverter
from pdfminer.layout import LAParams
from pdfminer.pdfpage import PDFPage
import io

def extract_text_from_pdf(pdf_path):
    resource_manager = PDFResourceManager()
    fake_file_handle = io.StringIO()
    converter = TextConverter(resource_manager, fake_file_handle, laparams=LAParams())
    page_interpreter = PDFPageInterpreter(resource_manager, converter)

    with open(pdf_path, 'rb') as fh:
        for page in PDFPage.get_pages(fh, caching=True, check_extractable=True):
            page_interpreter.process_page(page)

        text = fake_file_handle.getvalue()

    converter.close()
    fake_file_handle.close()

    if text:
        return text.strip()

# 调用示例
text = extract_text_from_pdf('example.pdf')
print(text)

在这个例子中，使用pdfminer库提取了example.pdf文件中的文本信息，并输出到了控制台。

其中也使用了pdfinterp、converter、layout和pdfpage这些模块，其中：

pdfinterp模块用于处理PDF文件；
converter模块用于将PDF文件转换为文本；
layout模块用于处理页面布局；
pdfpage模块用于处理PDF页面。

reportlab库

reportlab可以在Python中生成PDF文档，支持PDF文档中的各种元素，例如文本、图像、图形、表格等。reportlab库提供了丰富的API，允许开发人员使用Python代码快速构建高质量的PDF文档。

reportlab库有以下特点：

提供多种文本格式、颜色和字体选择
提供多种图像格式、缩放、剪裁等操作
支持图形元素，如线条、矩形、圆形、多边形等
支持表格，可以设置表格的大小、行列数、边框等
可以设置页面的大小、页边距等
支持页面链接和书签
支持水印和加密

下面是一个使用reportlab库创建PDF文档的示例：

from reportlab.pdfgen import canvas
from reportlab.lib.pagesizes import letter
from reportlab.lib.units import inch

# 创建一个PDF文档，页面大小为letter
pdf = canvas.Canvas("example.pdf", pagesize=letter)

# 设置字体
pdf.setFont("Helvetica", 12)

# 写入文本
text = "Hello, world!"
pdf.drawString(1 * inch, 10.5 * inch, text)

# 画一条线
pdf.line(1 * inch, 10.4 * inch, 6.5 * inch, 10.4 * inch)

# 画一个矩形
pdf.rect(1 * inch, 9.5 * inch, 1 * inch, 1 * inch)

# 添加一张图片
from reportlab.lib.utils import ImageReader
image = ImageReader("example.jpg")
pdf.drawImage(image, 2 * inch, 7.5 * inch, width=3 * inch, height=3 * inch)

# 保存PDF文档
pdf.save()

这个例子中，我们使用了canvas类创建了一个PDF文档，并在文档中写入了文本、画了一条线、一个矩形和一张图片。最后，我们调用了save方法保存了PDF文档。

pdfrw库

pdfrw也是一个用于处理PDF文件的Python库，它可以用来读取、编辑和创建PDF文件。

pdfrw的主要特点有：

可以读取和编辑现有的PDF文件，包括页面内容、书签、元数据、表单和注释等。
可以使用Python代码创建新的PDF文件，包括文本、图像、表格和注释等。
支持PDF的所有标准字体，也可以使用自定义字体。
提供了灵活的页面布局和样式控制功能。

接下来我们演示下用pdfrw库读取、编辑、创建PDF文件的示例。

读取PDF文件并输出文本内容：

import pdfrw

pdf = pdfrw.PdfReader('input.pdf')
for page in pdf.pages:
    print(page.Contents)

编辑PDF文件并保存：

import pdfrw

pdf = pdfrw.PdfReader('input.pdf')
for page in pdf.pages:
    page.Rotate = 90  # 旋转页面

pdfrw.PdfWriter().write('output.pdf', pdf)

创建新的PDF文件：

import pdfrw

from pdfrw import PdfDict, PdfName, PdfObject, PdfReader, PdfWriter
from pdfrw.buildxobj import pagexobj

width, height = 100, 100
page = pagexobj((0, 0, width, height))

xobj = PdfDict(
    Type = PdfName.XObject,
    Subtype = PdfName.Form,
    BBox = PdfObject([0, 0, width, height]),
    Resources = PdfDict(),
    stream = page.stream
)

pdf = PdfReader('input.pdf')
pdf.pages.append(xobj)

PdfWriter().write('output.pdf', pdf)

pdfrw的文档较为简单，但提供了大量示例代码和函数参数说明，易于上手。它的主要优势是对PDF格式的原生支持，能够更好地处理PDF文件的细节和特殊情况。