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