Python處理pdf檔案庫 - PyPDF2詳解

pythontab發表於2017-11-28

工作中可能會涉及處理pdf檔案,PyPDF2就是這樣一個庫, 使用它可以輕鬆的處理pdf檔案,它提供了讀、寫、分割、合併、檔案轉換等多種操作。官方地址:http://mstamy2.github.io/PyPDF2/

安裝

1. RPM式系統(Redhat、CentOS)

pip install pypdf2

2. DEB式系統(Debian、Ubuntu)以下任一

pip install pypdf2
apt install python-pypdf2

3. Windows

pip install pypdf2

使用

PyPDF2 包含了 PdfFileReader PdfFileMerger PageObject PdfFileWriter 四個常用的主要 Class。

簡單讀寫

from PyPDF2 import PdfFileReader, PdfFileWriter
readFile = 'read.pdf'
writeFile = 'write.pdf'
# 獲取一個 PdfFileReader 物件
pdfReader = PdfFileReader(open(readFile, 'rb'))
# 獲取 PDF 的頁數
pageCount = pdfReader.getNumPages()
print(pageCount)
# 返回一個 PageObject
page = pdfReader.getPage(i)
# 獲取一個 PdfFileWriter 物件
pdfWriter = PdfFileWriter()
# 將一個 PageObject 加入到 PdfFileWriter 中
pdfWriter.addPage(page)
# 輸出到檔案中
pdfWriter.write(open(writeFile, 'wb'))


合併分割 PDF

from PyPDF2 import PdfFileReader, PdfFileWriter
def split_pdf(infn, outfn):
    pdf_output = PdfFileWriter()
    pdf_input = PdfFileReader(open(infn, 'rb'))
    # 獲取 pdf 共用多少頁
    page_count = pdf_input.getNumPages()
    print(page_count)
    # 將 pdf 第五頁之後的頁面,輸出到一個新的檔案
    for i in range(5, page_count):
        pdf_output.addPage(pdf_input.getPage(i))
    pdf_output.write(open(outfn, 'wb'))
def merge_pdf(infnList, outfn):
    pdf_output = PdfFileWriter()
    for infn in infnList:
        pdf_input = PdfFileReader(open(infn, 'rb'))
        # 獲取 pdf 共用多少頁
        page_count = pdf_input.getNumPages()
        print(page_count)
        for i in range(page_count):
            pdf_output.addPage(pdf_input.getPage(i))
    pdf_output.write(open(outfn, 'wb'))
if __name__ == '__main__':
    infn = 'infn.pdf'
    outfn = 'outfn.pdf'
    split_pdf(infn, outfn)

其他命令

如果是要修改一個已有的 pdf 檔案,可以將 reader 的頁面新增到 writer 中:

pdfWriter.appendPagesFromReader(reader)

新增書籤:

pdfWriter.addBookmark(title, pagenum, parent=parent)


相關文章