Python应用——自定义函数：分割PDF文件函数

案例

将一个 pdf 文件按要求分割为几个部分。比如说一个pdf有20页，分成5个pdf文件，每个pdf文件包含4页。设计函数实现？

python代码

from pypdf2 import pdffilereader, pdffilewriter
# 分割 pdf 函数定义
def splitpdf(input_pdf, out_num, split_site):

    for i in range(out_num):
        with open(input_pdf, 'rb') as open_pdf, \
                open(str(i)+'.pdf', 'wb') as write_pdf:

            pdfreader = pdffilereader(open_pdf)

            pdfwriter = pdffilewriter()
            for j in range(i*split_site, (i+1)*split_site):
                page = pdfreader.getpage(j)
                pdfwriter.addpage(page)

            pdfwriter.write(write_pdf)

# 测试函数
splitpdf('inputfile.pdf', 5, 4)
# 参数：输入pdf文件，分割份数（5个pdf文件），分割页数（每个pdf4页）

函数讲解

本函数是自己测试通过的函数，还有待优化。输入参数有，将要分割的pdf文件，分割为几个pdf文件，每个pdf文件页数。

测试结果
原来只有一个：inputfile.pdf 文件
运行之后为：0.pdf; 1.pdf; 2.pdf; 3.pdf; 4.pdf