工作中可能會涉及處理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)