使用python将pdf拆分为多个不同页长的pdf。

import os
from PyPDF2 import PdfFileReader, PdfFileWriter

def split_pdf(input_path, output_dir, page_lengths):
    # 读取输入PDF文件
    with open(input_path, 'rb') as file:
        pdf = PdfFileReader(file)
        total_pages = pdf.getNumPages()

        # 确保页长列表的长度与总页数相同
        if len(page_lengths) != total_pages:
            raise ValueError("页长列表的长度与总页数不匹配")

        # 拆分PDF
        for i in range(total_pages):
            page_length = page_lengths[i]

            # 创建一个新的PDF写入器
            writer = PdfFileWriter()
            writer.addPage(pdf.getPage(i))

            # 构造输出文件路径
            output_path = os.path.join(output_dir, f"page_{i+1}_length_{page_length}.pdf")

            # 写入拆分后的PDF文件
            with open(output_path, 'wb') as output_file:
                writer.write(output_file)

            print(f"已成功拆分第 {i+1} 页，页长为 {page_length} 的PDF文件：{output_path}")

# 示例用法
input_pdf = 'input.pdf'  # 输入PDF文件路径
output_directory = 'output'  # 输出目录路径
page_lengths = [1, 2, 3, 4]  # 页长列表，与总页数相同

split_pdf(input_pdf, output_directory, page_lengths)