当前位置：首页 > news >正文

Python实现PDF按页分割：灵活拆分文档的技术指南

news 2025/7/26 19:56:57

Python实现PDF按页分割：灵活拆分文档的技术指南

PDF文件处理是日常工作中的常见需求，特别是当我们需要将大型PDF文档拆分为多个部分时。本文将介绍如何使用Python创建一个灵活的PDF分割工具，能够根据用户指定的页数范围任意分割文档。

需求分析

从实际案例出发，今天一个朋友给向我求助帮忙，将一份PDF文档（75页），需要将其分割为三个部分：

第1部分：第1页（申请表）
第2部分：第2-9页（身份证明和医疗证明）
第3部分：第10页到最后（费用明细）

这个问题用Python语言开发一个程序轻松搞定。不仅能满足这个特定需求，还能处理任意页数范围的分割。

技术方案

工具选择

PyPDF2：轻量级PDF处理库，适合基本PDF操作
os：处理文件和目录路径
argparse：解析命令行参数

安装依赖

pip install PyPDF2

完整代码实现

import os
import argparse
from PyPDF2 import PdfReader, PdfWriterdef split_pdf(input_path, output_folder, ranges):"""将PDF文件按指定页数范围分割参数:input_path (str): 输入PDF文件路径output_folder (str): 输出文件夹路径ranges (list): 分割范围列表，格式为[(起始页, 结束页), ...]"""# 确保输出文件夹存在if not os.path.exists(output_folder):os.makedirs(output_folder)# 获取文件名（不含扩展名）file_name = os.path.splitext(os.path.basename(input_path))[0]try:with open(input_path, 'rb') as file:reader = PdfReader(file)total_pages = len(reader.pages)print(f"开始处理: {input_path}")print(f"总页数: {total_pages}")# 处理每个分割范围for i, (start, end) in enumerate(ranges):# 处理结束页为-1（表示到文档末尾）的情况if end == -1:end = total_pages# 验证页码范围if start < 1 or end > total_pages or start > end:print(f"⚠️ 警告: 无效的页码范围 [{start}, {end}]，跳过")continue# 创建PDF写入对象writer = PdfWriter()# 添加指定范围的页面（PyPDF2使用0-based索引）for page_num in range(start - 1, end):writer.add_page(reader.pages[page_num])# 生成输出文件名part_name = f"{file_name}_part{i+1}_p{start}-{end}.pdf"output_path = os.path.join(output_folder, part_name)# 写入文件with open(output_path, 'wb') as output_file:writer.write(output_file)print(f"✅ 已创建: {part_name} (页数: {start}-{end})")print(f"分割完成! 文件保存在: {output_folder}")except Exception as e:print(f"❌ 处理过程中发生错误: {str(e)}")def parse_ranges(range_str):"""解析页数范围字符串参数:range_str (str): 页数范围字符串，格式如 "1,2-9,10-"返回:list: 解析后的页数范围列表"""ranges = []parts = range_str.split(',')for part in parts:if '-' in part:start_end = part.split('-')start = int(start_end[0].strip())# 处理结束页为空的特殊情况（表示到文档末尾）if start_end[1].strip() == '':end = -1  # 使用-1表示到文档末尾else:end = int(start_end[1].strip())else:# 单个页码start = int(part.strip())end = startranges.append((start, end))return rangesif __name__ == "__main__":# 设置命令行参数解析parser = argparse.ArgumentParser(description="PDF文件分割工具 - 按指定页数范围分割PDF文档",formatter_class=argparse.RawTextHelpFormatter)parser.add_argument('input', help='输入PDF文件路径')parser.add_argument('ranges', help='页数范围，格式如 "1,2-9,10-"\n''示例: "1" - 仅第一页\n''       "1-5" - 1到5页\n''       "10-" - 从第10页到文档末尾\n''       "1,3-5,7-" - 多个范围')parser.add_argument('-o', '--output', default='pdf_split_output',help='输出文件夹 (默认: pdf_split_output)')args = parser.parse_args()# 解析页数范围try:ranges = parse_ranges(args.ranges)print(f"解析的分割范围: {ranges}")# 执行PDF分割split_pdf(args.input, args.output, ranges)except ValueError:print("❌ 错误: 页数范围格式无效。请使用格式如 '1,2-9,10-'")except Exception as e:print(f"❌ 发生错误: {str(e)}")

使用说明

基本用法

python pdf_splitter.py 输入文件.pdf "页数范围" [-o 输出文件夹]

示例命令

分割为三个部分（第1页、第2-9页、第10页到最后）：

python pdf_splitter.py "吉林省镇赉县*.pdf" "1,2-9,10-" -o 分割结果

仅提取特定页面：

# 提取第5页
python pdf_splitter.py input.pdf "5"# 提取第10-15页
python pdf_splitter.py input.pdf "10-15"# 提取从第20页到文档末尾
python pdf_splitter.py input.pdf "20-"

提取多个不连续范围：

# 提取封面(1)、目录(3-4)和正文(6-)
python pdf_splitter.py book.pdf "1,3-4,6-"

输出示例

解析的分割范围: [(1, 1), (2, 9), (10, -1)]
开始处理: 吉林省镇赉县*.pdf
总页数: 75
✅ 已创建: 吉林省镇赉县*_part1_p1-1.pdf (页数: 1-1)
✅ 已创建: 吉林省镇赉县*_part2_p2-9.pdf (页数: 2-9)
✅ 已创建: 吉林省镇赉县*_part3_p10-75.pdf (页数: 10-75)
分割完成! 文件保存在: 分割结果