如何在Python中编辑PDF文件

有时候我们需要在Python程序中对PDF文件进行编辑，例如添加水印，合并多个PDF文件，提取页面等。本文将介绍几种在Python中编辑PDF文件的方法。

1. PyPDF2

PyPDF2是一个Python库，可以用于操作PDF文件，包括合并，分割，提取文本，加密和解密PDF文件等。下面是使用PyPDF2添加文字水印的示例代码：

import PyPDF2

# 打开PDF文件
pdf_file = open('example.pdf', 'rb')
pdf_reader = PyPDF2.PdfFileReader(pdf_file)

# 添加文字水印
pdf_writer = PyPDF2.PdfFileWriter()
for page_num in range(pdf_reader.numPages):
    page = pdf_reader.getPage(page_num)
    page.mergePage(pdf_reader.getPage(page_num))
    pdf_writer.addPage(page)
pdf_writer.addBookmark('Watermark', 0)
pdf_writer.encrypt('password')
result_pdf = open('result.pdf', 'wb')
pdf_writer.write(result_pdf)
result_pdf.close()
pdf_file.close()

2. pdfminer

pdfminer是一个有用的Python库，用于提取PDF文本和元数据。可以使用pdfminer提取PDF中的文本或其他元素，如图像，链接等。下面是使用pdfminer提取PDF中的文本的示例代码：

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

def extract_pdf_text(pdf_file_path):
    resource_manager = PDFResourceManager()
    out_file = io.StringIO()
    la_params = LAParams()
    device = TextConverter(
        resource_manager, out_file, laparams=la_params, 
        imagewriter=None, codec='utf-8', 
        pageno=1, showpageno=False, 
        laparams_mode=False
        )
    with open(pdf_file_path, 'rb') as fp:
        interpreter = PDFPageInterpreter(resource_manager, device)
        for page in PDFPage.get_pages(fp):
            interpreter.process_page(page)
    return out_file.getvalue()

3. pdfrw

pdfrw是一个Python库，可用于编辑现有PDFs，例如添加页码，提取页面等。下面是使用pdfrw添加页码的示例代码：

from pdfrw import PdfReader, PdfWriter, PageMerge

def add_page_number(input_path, output_path):
    pdf = PdfReader(input_path)
    writer = PdfWriter(output_path)
    page_number = 0
    
    for page in pdf.pages:
        page_number += 1
        PageMerge(page).add(
            PageMerge().add(
                pdf.pages[0]
                ),
            prepend=True
            )
        writer.addpage(page)
        
    writer.write()

这些都是在Python中编辑PDF文件的示例方法。当然在Python中还有许多其他库可供使用，具体取决于你的具体需求。