帮我写一个指定页数分割pdf的python的代码

在Python中，我们可以使用第三方库如`PyPDF2`来操作PDF文件，包括分页。以下是一个基本的示例，展示如何将一个PDF文件分割成指定页数的独立PDF文档：

import PyPDF2

def split_pdf(input_file, output_prefix, start_page, end_page):
    # 打开原始PDF文件
    with open(input_file, 'rb') as reader:
        pdf = PyPDF2.PdfFileReader(reader)

        # 确保开始和结束页有效
        if start_page < 0 or end_page > pdf.getNumPages() or start_page >= end_page:
            print("Invalid page range.")
            return

        # 创建一个新的PDF writer
        writer = PyPDF2.PdfFileWriter()

        # 分割并添加页面到新文档
        for page_num in range(start_page, end_page + 1):
            writer.addPage(pdf.getPage(page_num - 1))  # 减一是因为索引从0开始

        # 将分割后的页面写入新的PDF文件
        for i in range(start_page, end_page + 1):
            output_file = f"{output_prefix}_{i}.pdf"
            with open(output_file, 'wb') as output:
                writer.write(output)
        print(f"Successfully split PDF into pages {start_page}-{end_page}.")

# 调用函数，例如把第5页到第8页分割出来
split_pdf('input.pdf', 'output_', 4, 7)  # 输出前缀，起始页，结束页