1. 程式人生 > >Python提取PDF內容(文字、影象、線條等)

Python提取PDF內容(文字、影象、線條等)

使用Python抽取PDF檔案內容,包括文字、影象、線條等物件

摘要:這篇文章主要介紹如何使用Python【3.6版本】中的PDFminer3k 模組來抽取PDF內容,包括文字、影象、曲線等。

作者:yooongchun
微信公眾號:yooongchun小屋
這裡寫圖片描述

1.安裝PDFminer3k

使用pip 命令安裝

pip install pdfminer3k

2.編寫測試

  • 你可以在這裡獲得官方參考:PDFMiner
  • 如果你不喜歡看英文的官方文件,這裡的翻譯也許對你有幫助:中文PDFMiner文件

下面的程式,我拓展了官方給出的例子,你可以通過這個例子統計出來你的pdf檔案一共包含哪些內容,比如文字框,曲線,圖片等

#!/usr/bin/python
# -*- coding: utf-8 -*-

__author__ = 'yooongchun'

import sys
import importlib
importlib.reload(sys)

from pdfminer.pdfparser import PDFParser,PDFDocument
from pdfminer.pdfinterp import PDFResourceManager, PDFPageInterpreter
from pdfminer.converter import PDFPageAggregator
from pdfminer.layout import
* from pdfminer.pdfinterp import PDFTextExtractionNotAllowed ''' 解析pdf檔案,獲取檔案中包含的各種物件 ''' # 解析pdf檔案函式 def parse(pdf_path): fp = open(pdf_path, 'rb') # 以二進位制讀模式開啟 # 用檔案物件來建立一個pdf文件分析器 parser = PDFParser(fp) # 建立一個PDF文件 doc = PDFDocument() # 連線分析器 與文件物件 parser.set_document(doc) doc.set_parser(parser) # 提供初始化密碼
# 如果沒有密碼 就建立一個空的字串 doc.initialize() # 檢測文件是否提供txt轉換,不提供就忽略 if not doc.is_extractable: raise PDFTextExtractionNotAllowed else: # 建立PDf 資源管理器 來管理共享資源 rsrcmgr = PDFResourceManager() # 建立一個PDF裝置物件 laparams = LAParams() device = PDFPageAggregator(rsrcmgr, laparams=laparams) # 建立一個PDF直譯器物件 interpreter = PDFPageInterpreter(rsrcmgr, device) # 用來計數頁面,圖片,曲線,figure,水平文字框等物件的數量 num_page, num_image, num_curve, num_figure, num_TextBoxHorizontal = 0, 0, 0, 0, 0 # 迴圈遍歷列表,每次處理一個page的內容 for page in doc.get_pages(): # doc.get_pages() 獲取page列表 num_page += 1 # 頁面增一 interpreter.process_page(page) # 接受該頁面的LTPage物件 layout = device.get_result() for x in layout: if isinstance(x,LTImage): # 圖片物件 num_image += 1 if isinstance(x,LTCurve): # 曲線物件 num_curve += 1 if isinstance(x,LTFigure): # figure物件 num_figure += 1 if isinstance(x, LTTextBoxHorizontal): # 獲取文字內容 num_TextBoxHorizontal += 1 # 水平文字框物件增一 # 儲存文字內容 with open(r'test.txt', 'a') as f: results = x.get_text() f.write(results + '\n') print('物件數量:\n','頁面數:%s\n'%num_page,'圖片數:%s\n'%num_image,'曲線數:%s\n'%num_curve,'水平文字框:%s\n' %num_TextBoxHorizontal) if __name__ == '__main__': pdf_path = r'C:\Users\fanyu\Desktop\pdf\test.pdf' parse(pdf_path)

其實在上面的layout 物件中有更多的內容可提取,這個自己按需來寫就好,然後對曲線,文字框等物件,都會有位置屬性,可直接獲取,自己debug 檢視以下物件屬性獲取即可。