使用python将大型PDF文件拆分为单个PDF

import os
from PyPDF2 import PdfFileReader, PdfFileWriter

def split_pdf(input_path, output_dir):
    # 打开大型PDF文件
    with open(input_path, 'rb') as file:
        pdf = PdfFileReader(file)
        # 获取PDF文件的总页数
        total_pages = pdf.getNumPages()

        # 遍历每一页，将其保存为单独的PDF文件
        for page_number in range(total_pages):
            # 创建一个新的PDF文件
            output_pdf = PdfFileWriter()
            # 将当前页添加到新的PDF文件中
            output_pdf.addPage(pdf.getPage(page_number))

            # 构造输出文件路径
            output_path = os.path.join(output_dir, f'page_{page_number + 1}.pdf')
            # 保存新的PDF文件
            with open(output_path, 'wb') as output_file:
                output_pdf.write(output_file)

# 调用函数进行拆分
input_path = 'path/to/large_pdf.pdf'
output_dir = 'path/to/output_directory'
split_pdf(input_path, output_dir)