用Python切割PDF代码

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

from PyPDF2 import PdfFileWriter, PdfFileReader

# 读入文件
pdf_file = PdfFileReader(open("full.pdf", "rb"))

# sp - 开始页（包含）;  ep - 结束页（包含）
def save_selected_pages(sp, ep, pdf_file_, out_name):
    output = PdfFileWriter()
    
    for i in range(sp-1, ep):
        output.addPage(pdf_file_.getPage(i))
    
    outputStream = open(out_name, "wb")
    output.write(outputStream)
    outputStream.close()
    
# 例如截取第2到6页（包含）
save_selected_pages(2, 6, pdf_file, "1.pdf")