1. 程式人生 > >Python應用——自定義函數:分割PDF文件函數

Python應用——自定義函數:分割PDF文件函數

ade site 要求 import 輸入 open fread inpu pen

案例

將一個 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文件函數