Python使用pypdf实现按自定义页码拆分PDF文件

pip install pypdf

from pypdf import PdfReader, PdfWriter
import os

def split_pdf_by_page_ranges(input_pdf, output_folder, ranges):
    reader = PdfReader(input_pdf)

    # 如果输出目录不存在，则创建
    if not os.path.exists(output_folder):
        os.makedirs(output_folder)

    # 遍历页码范围
    for idx, (start_page, end_page) in enumerate(ranges, 1):
        writer = PdfWriter()

        # 页码从 0 开始，用户输入通常是从 1 开始
        for page_num in range(start_page - 1, end_page):
            if page_num < len(reader.pages):
                writer.add_page(reader.pages[page_num])
            else:
                print(f"Warning: page {page_num + 1} out of range.")

        # 输出文件名
        output_filename = f"{output_folder}/chapter_{idx}_pages_{start_page}-{end_page}.pdf"
        with open(output_filename, "wb") as output_file:
            writer.write(output_file)

        print(f"Saved {output_filename}")

# ✏️ 自定义你的分章节页码（格式：[(开始页, 结束页), ...]）
page_ranges = [
    (1, 34),   # 序
    (35, 50),  # 第一章
    (51, 73),  # 第二章
    (74, 93),  # 第三章
    (94, 118), # 第四章
    (119, 152),# 第五章
    (153, 166),# 第六章
    (167, 183),# 第七章
    (184, 206),# 第八章
    (207, 230),# 第九章
    (231, 251) # 第十章
]

# 调用函数拆分 PDF
split_pdf_by_page_ranges("input.pdf", "./output_manual_split", page_ranges)

reader = PdfReader(input_pdf)

if not os.path.exists(output_folder):
    os.makedirs(output_folder)

for page_num in range(start_page - 1, end_page):
    writer.add_page(reader.pages[page_num])

with open(output_filename, "wb") as output_file:
    writer.write(output_file)