网站首页 文章专栏 使用pdfminer提取pdf数据.md
安装
pip3 install pdfminer3k
网络上的代码都是旧版本不可运行,参考 stackoverflow.com 修改了下面的版本,可以正常运行。但问题就是某些 pdf 就是提取不出来文字,而且明明就是可编辑的 pdf 文件。最终采取的方案是一个一个 pdf 文件打开,然后全选、复制、粘贴,比较顺畅。
#--coding:utf-8--
import importlib
import sys
import time
importlib.reload(sys)
time1 = time.time()
from pdfminer.pdfparser import PDFParser
from pdfminer.pdfdocument import PDFDocument
from pdfminer.pdfpage import PDFPage
from pdfminer.pdfinterp import PDFResourceManager, PDFPageInterpreter
from pdfminer.converter import PDFPageAggregator
from pdfminer.layout import LTTextBoxHorizontal, LAParams
# from pdfminer.pdfinterp import PDFTextExtractionNotAllowed
def parse(PDF_path,TXT_path):
'''解析PDF文本,并保存到TXT文件中'''
fp = open(PDF_path, 'rb')
# 用文件对象创建一个PDF文档分析器
parser = PDFParser(fp)
# 创建一个PDF文档
# doc = PDFDocument()
doc = PDFDocument(parser)
# 连接分析器,与文档对象
parser.set_document(doc)
# doc.set_parser(parser)
# 提供初始化密码,如果没有密码,就创建一个空的字符串
# doc.initialize()
# 检测文档是否提供txt转换,不提供就忽略
if not doc.is_extractable:
# raise PDFTextExtractionNotAllowed
print("Extract not allowed")
else:
# 创建PDF,资源管理器,来共享资源
rsrcmgr = PDFResourceManager()
# 创建一个PDF设备对象
laparams = LAParams()
device = PDFPageAggregator(rsrcmgr, laparams=laparams)
# 创建一个PDF解释其对象
interpreter = PDFPageInterpreter(rsrcmgr, device)
# 循环遍历列表,每次处理一个page内容
# doc.get_pages() 获取page列表
# for page in doc.get_pages():
for page in PDFPage.create_pages(doc):
interpreter.process_page(page)
# 接受该页面的LTPage对象
layout = device.get_result()
# 这里layout是一个LTPage对象 里面存放着 这个page解析出的各种对象
# 一般包括LTTextBox, LTFigure, LTImage, LTTextBoxHorizontal 等等
# 想要获取文本就获得对象的text属性,
for x in layout:
if (isinstance(x, LTTextBoxHorizontal)):
with open(TXT_path, 'a') as f:
results = x.get_text()
print(results)
f.write(results + "\n")
if __name__ == '__main__':
PDF_path = '建设银行.pdf'
TXT_path = '321.txt'
parse(PDF_path,TXT_path)
time2 = time.time()
print("总共消耗时间为:", time2 - time1)