PDF文件页面提取操作小指南

在本教程中,我们将使用 PyPDF2 库中的 PdfWriterPdfReader 类,以及 Python 的 pathlib 模块,对 PDF 文件进行页面提取和分割操作。我们将以名为 “Pride_and_Prejudice.pdf” 的 PDF 文件为例,演示如何实现以下功能:

  1. 提取最后一页并保存为 last_page.pdf
  2. 提取偶数索引的页面并保存为 every_other_page.pdf
  3. 将 PDF 文件分成两个部分:前 150 页和后面的页面,分别保存为 part_1.pdfpart_2.pdf

环境准备

首先确保安装了所需的库。如果没有安装,可以通过以下命令安装 PyPDF2

pip install PyPDF2

实现代码

以下是实现上述功能的完整代码:

from PyPDF2 import PdfReader, PdfWriter
from pathlib import Path

# 定义文件路径
# 定义PDF文件路径
pdf_path = Path(__file__).parent / 'practice_files' / 'Pride_and_Prejudice.pdf'
home_directory = Path.home()

# 读取 PDF 文件
reader = PdfReader(str(pdf_file_path))

# 提取最后一页并保存为 last_page.pdf
writer_last_page = PdfWriter()
last_page = reader.pages[-1]
writer_last_page.add_page(last_page)
with open(home_directory / "last_page.pdf", "wb") as last_page_pdf:
    writer_last_page.write(last_page_pdf)

# 提取偶数索引的页面(0, 2, 4, ...)并保存为 every_other_page.pdf
writer_even_pages = PdfWriter()
for i in range(0, len(reader.pages), 2):
    writer_even_pages.add_page(reader.pages[i])
with open(home_directory / "every_other_page.pdf", "wb") as even_pages_pdf:
    writer_even_pages.write(even_pages_pdf)

# 将页面分成两部分
split_index = 151
pdf_parts = [PdfWriter(), PdfWriter()]

# 添加页面到对应的PdfWriter对象
for i, page in enumerate(pdf_reader.pages):
    print(i // split_index)
    pdf_parts[i // split_index].add_page(page)

# 定义保存PDF的函数
def save_pdf(writer, filename):
    with Path(Path.home() / filename).open('wb') as f:
        writer.write(f)

# 保存两个PDF文件
save_pdf(pdf_parts[0], 'part_1.pdf')
save_pdf(pdf_parts[1], 'part_2.pdf')

print("PDF 文件处理完成!")

代码解释

  1. 导入模块: 我们先导入所需的 PdfReader, PdfWriterPath 模块。
  2. 定义文件路径: 使用 pathlib.Path 设置 PDF 文件路径和家目录。
  3. 读取 PDF 文件: 使用 PdfReader 读取 PDF 文件内容。
  4. 提取最后一页: 通过 reader.pages[-1] 获取最后一页,并用 PdfWriter 写入新的 PDF 文件。
  5. 提取偶数索引页面: 使用循环遍历偶数索引,将其添加到新的 PdfWriter 对象中。
  6. 分割 PDF 文件: 创建两个写入对象,将前 150 页和剩余页分别添加到不同的 PdfWriter 中,并保存为 part_1.pdfpart_2.pdf

运行结果

运行以上代码后,你将在用户的家目录下找到以下文件:

  • last_page.pdf: 最后一页
  • every_other_page.pdf: 偶数索引页
  • part_1.pdf: 前 150 页
  • part_2.pdf: 剩余的页面

这样,我们就成功地对 PDF 文件进行了页面提取和分割操作!快快体验吧!!欢迎点赞、关注、转发!!!

上一篇:Linux下网络转发功能


下一篇:Transforms(一)