在python中使用PDFMiner从PDF文件中提取文本?

Posted

技术标签:

【中文标题】在python中使用PDFMiner从PDF文件中提取文本?【英文标题】:Extracting text from a PDF file using PDFMiner in python? 【发布时间】:2014-12-17 02:54:44 【问题描述】:

我正在寻找有关如何使用 PDFMiner 和 Python 从 PDF 文件中提取文本的文档示例。

看起来 PDFMiner 更新了他们的 API,我发现的所有相关示例都包含过时的代码(类和方法已更改)。我发现使从 PDF 文件中提取文本的任务更容易的库正在使用旧的 PDFMiner 语法,所以我不知道该怎么做。

事实上,我只是在查看源代码,看看我是否能弄清楚。

【问题讨论】:

请查看 ***.com/help/how-to-ask 和 ***.com/help/mcve 并更新您的答案,使其格式更好并符合指南。 您使用的是哪个 Python 发行版,2.7.x 还是 3.x.x?需要注意的是,作者明确详细说明了PDFminer 不适用于 Python 3.x.x。这可能是您收到import 错误的原因。如果是这样,您应该使用pdfminer3k,因为它是该库的常规 Python 3 导入。 @Nanashi,对不起,我忘了添加我的 Python 版本。这是2.7,所以这不是问题。我一直在查看源代码,看起来他们重组了一些东西,这就是导入中断的原因。我也找不到 PDFMiner 的任何文档,或者我只是在解决这个问题:( 我刚刚从 GitHub 安装了PDFminer,它可以正常导入。能否请您发布您的代码并发布您的完整错误回溯? @Nanashi,就像我在最初的问题中所说的那样,依赖 PDFMiner 的库在完成导入之前会中断以及我能找到的任何示例。这不是 PDFMiner 问题。这是我在寻找文档,或如何使用 PDFMiner 的示例。我能找到的一切都是使用 PDFMiner 的旧语法。为了清楚起见,我继续编辑了我的问题。我想我让它变得比它需要的更混乱。对此感到抱歉。 【参考方案1】:

这是一个使用当前版本的 PDFMiner(2016 年 9 月)从 PDF 文件中提取文本的工作示例

from pdfminer.pdfinterp import PDFResourceManager, PDFPageInterpreter
from pdfminer.converter import TextConverter
from pdfminer.layout import LAParams
from pdfminer.pdfpage import PDFPage
from io import StringIO

def convert_pdf_to_txt(path):
    rsrcmgr = PDFResourceManager()
    retstr = StringIO()
    codec = 'utf-8'
    laparams = LAParams()
    device = TextConverter(rsrcmgr, retstr, codec=codec, laparams=laparams)
    fp = open(path, 'rb')
    interpreter = PDFPageInterpreter(rsrcmgr, device)
    password = ""
    maxpages = 0
    caching = True
    pagenos=set()

    for page in PDFPage.get_pages(fp, pagenos, maxpages=maxpages, password=password,caching=caching, check_extractable=True):
        interpreter.process_page(page)

    text = retstr.getvalue()

    fp.close()
    device.close()
    retstr.close()
    return text

PDFMiner 的结构最近发生了变化,因此这应该适用于从 PDF 文件中提取文本。

编辑:截至 2018 年 6 月 7 日仍在工作。在 Python 版本 3.x 中验证

编辑:该解决方案适用于 2019 年 10 月 3 日的 Python 3.7。我使用了 2018 年 11 月发布的 Python 库 pdfminer.six

【讨论】:

工作正常,但是,如何处理例如名称中的空格?假设我有一个包含 4 列的 pdf,其中我在一个列中有名字和姓氏,现在它被解析为一行中的名字和一行中的姓氏,这是一个示例 docdro.id/rRyef3x 当前使用此代码出现导入错误:ImportError: No module named 'pdfminer.pdfpage' 感谢它在 python v2.7.12 和 ubuntu 16.04 上工作,但最好加载编码为 utf-8 的 pdf 文档,因为我的示例 pdf 有一些编码问题,所以在编码后试试这个使用 utf-8 即可解决问题...import sys reload(sys) sys.setdefaultencoding('utf-8') @DuckPuncher,它现在还在工作吗?我必须将 file(path, 'rb') 更改为 `open(path, 'rb') 才能让我的工作。 仍然为 Python3.7 用户工作。安装 pdfminer.six==20181108 包。迄今为止我的案例的最佳解决方案,我比较了许多解决方案。【参考方案2】:

来自 DuckPuncher 的出色回答,对于 Python3,请确保安装 pdfminer2 并执行以下操作:

import io

from pdfminer.pdfinterp import PDFResourceManager, PDFPageInterpreter
from pdfminer.converter import TextConverter
from pdfminer.layout import LAParams
from pdfminer.pdfpage import PDFPage


def convert_pdf_to_txt(path):
    rsrcmgr = PDFResourceManager()
    retstr = io.StringIO()
    codec = 'utf-8'
    laparams = LAParams()
    device = TextConverter(rsrcmgr, retstr, codec=codec, laparams=laparams)
    fp = open(path, 'rb')
    interpreter = PDFPageInterpreter(rsrcmgr, device)
    password = ""
    maxpages = 0
    caching = True
    pagenos = set()

    for page in PDFPage.get_pages(fp, pagenos, maxpages=maxpages,
                                  password=password,
                                  caching=caching,
                                  check_extractable=True):
        interpreter.process_page(page)



    fp.close()
    device.close()
    text = retstr.getvalue()
    retstr.close()
    return text

【讨论】:

它对我不起作用:ModuleNotFoundError: No module named 'pdfminer.pdfpage' 我正在使用 python 3.6 @Atti,以防万一,请确保您已安装 pdfminer2,因为还有另一个软件包 pdfminer(我讨厌这个)。在进行 pip3 freeze 时,它​​适用于 pdfminer2==20151206 版本。 感谢我终于成功了,我从 conda forge 安装了 pdfminer.six 对于 Python 3,推荐使用 pdfminer.six 包 - github.com/pdfminer/pdfminer.six 这仍然是最新的。我收到了同样的ImportError: 消息【参考方案3】:

此代码使用 pdfminer for python 3 (pdfminer-20191125) 测试

from pdfminer.layout import LAParams
from pdfminer.converter import PDFPageAggregator
from pdfminer.pdfinterp import PDFResourceManager
from pdfminer.pdfinterp import PDFPageInterpreter
from pdfminer.pdfpage import PDFPage
from pdfminer.layout import LTTextBoxHorizontal

def parsedocument(document):
    # convert all horizontal text into a lines list (one entry per line)
    # document is a file stream
    lines = []
    rsrcmgr = PDFResourceManager()
    laparams = LAParams()
    device = PDFPageAggregator(rsrcmgr, laparams=laparams)
    interpreter = PDFPageInterpreter(rsrcmgr, device)
    for page in PDFPage.get_pages(document):
            interpreter.process_page(page)
            layout = device.get_result()
            for element in layout:
                if isinstance(element, LTTextBoxHorizontal):
                    lines.extend(element.get_text().splitlines())
    return lines

【讨论】:

我有可以使用 Nitro Pro 工具转换的 PDF 文件。但是,当我尝试使用此处发布的代码转换相同的 PDF 时,我得到的输出表明存在权限错误。这是输出: ('来自 SAGE 社会科学合集。保留所有权利。\n\n\x0c\x0c\x0c\x0c\x0c\x0c\x0c\x0c\x0c\x0c\x0c\x0c\x0c\x0c \x0c\x0c\x0c\x0c\x0c\x0c\x0c\x0c\x0c\x0c\x0c\x0c\x0c\x0c\x0c') 文件流是什么意思? @Vincent 以 open(file,'rb') 作为流:[...] 您是否设法将此文件作为表格/熊猫理想地获得? groupe-psa.com/en/publication/monthly-world-sales-march-2020【参考方案4】:

完全披露,我是pdfminer.six 的维护者之一。它是用于 python 3 的 pdfminer 的社区维护版本。

现在,它有多个 api 用于从 PDF 中提取文本,具体取决于您的需要。在幕后,所有这些 api 都使用相同的逻辑来解析和分析布局。

(所有示例都假定您的 PDF 文件名为 example.pdf

命令行

如果您只想提取文本一次,可以使用命令行工具 pdf2txt.py:

$ pdf2txt.py example.pdf

高级 API

如果你想用 Python 提取文本(属性),你可以使用高级 api。如果您想以编程方式从 PDF 中提取信息,这种方法是首选解决方案。

from pdfminer.high_level import extract_text

# Extract text from a pdf.
text = extract_text('example.pdf')

# Extract iterable of LTPage objects.
pages = extract_pages('example.pdf')

可组合的 api

还有一个可组合的 api,它在处理生成的对象时提供了很大的灵活性。例如,它允许您创建自己的布局算法。其他答案中建议使用此方法,但仅当您需要自定义某些组件时,我才会推荐此方法。

from io import StringIO

from pdfminer.converter import TextConverter
from pdfminer.layout import LAParams
from pdfminer.pdfdocument import PDFDocument
from pdfminer.pdfinterp import PDFResourceManager, PDFPageInterpreter
from pdfminer.pdfpage import PDFPage
from pdfminer.pdfparser import PDFParser

output_string = StringIO()
with open('example.pdf', 'rb') as in_file:
    parser = PDFParser(in_file)
    doc = PDFDocument(parser)
    rsrcmgr = PDFResourceManager()
    device = TextConverter(rsrcmgr, output_string, laparams=LAParams())
    interpreter = PDFPageInterpreter(rsrcmgr, device)
    for page in PDFPage.create_pages(doc):
        interpreter.process_page(page)

print(output_string.getvalue())

类似的问答here。我会尽量让它们保持同步。

【讨论】:

【参考方案5】:

这在 2020 年 5 月使用 Python3 中的 PDFminer 6 工作。

安装包

$ pip install pdfminer.six

导入包

from pdfminer.high_level import extract_text

使用保存在磁盘上的 PDF

text = extract_text('report.pdf')

或者:

with open('report.pdf','rb') as f:
    text = extract_text(f)

使用内存中已有的 PDF

如果 PDF 已在内存中,例如,如果使用 requests 库从 Web 检索,则可以使用 io 库将其转换为流:

import io

response = requests.get(url)
text = extract_text(io.BytesIO(response.content))

与 PyPDF2 相比的性能和可靠性

PDFminer.six 比 PyPDF2(某些类型的 PDF 失败)更可靠,尤其是 PDF 版本 1.7

但是,使用 PDFminer.six 提取文本的速度比 PyPDF2 慢 6 倍。

我在 15" MBP (2018) 上使用 timeit 定时提取文本,仅使用 10 页 PDF 定时提取功能(无文件打开等),并得到以下结果:

PDFminer.six: 2.88 sec
PyPDF2:       0.45 sec

pdfminer.six 占用空间也很大,需要 pycryptodome,它需要安装 GCC 和其他东西,将 Alpine Linux 上的最小安装 docker 映像从 80 MB 提升到 350 MB。 PyPDF2 没有明显的存储影响。

【讨论】:

以上是关于在python中使用PDFMiner从PDF文件中提取文本?的主要内容,如果未能解决你的问题,请参考以下文章

如何使用 pdfminer 作为库

用pdfminer.six打开一个PDF,从一个URL中打开。

用PDFMiner从PDF中提取文本文字

如何使用 pdfminer3k 读取 pdf 文件?

如何利用Python抓取PDF中的某些内容

如何在 Python 中解锁“安全”(读保护)PDF?