码迷,mamicode.com
首页 > 编程语言 > 详细

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

时间:2018-12-02 22:33:36      阅读:158      评论:0      收藏:0      [点我收藏+]

标签:优化   add   讲解   for   文件包含   range   结果   def   open   

案例

将一个 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

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

标签:优化   add   讲解   for   文件包含   range   结果   def   open   

原文地址:https://www.cnblogs.com/brightyuxl/p/10055301.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!