从pd提取表

2024-04-16 10:40:42 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图从这个PDF中的表中获取数据。我试过pdfminer和pypdf,不过运气不太好,我无法真正从表中获取数据。

这是其中一张桌子的样子: enter image description here

如您所见,有些列用“x”标记。我正试着把这张表列成一个对象列表。

这是目前为止的代码,我现在正在使用pdfminer。

# pdfminer test
from pdfminer.pdfdocument import PDFDocument
from pdfminer.pdfparser import PDFParser
from pdfminer.pdfinterp import PDFResourceManager, PDFPageInterpreter
from pdfminer.pdfdevice import PDFDevice, TagExtractor
from pdfminer.pdfpage import PDFPage, PDFTextExtractionNotAllowed
from pdfminer.converter import XMLConverter, HTMLConverter, TextConverter, PDFPageAggregator
from pdfminer.cmapdb import CMapDB
from pdfminer.layout import LAParams, LTTextBox, LTTextLine, LTFigure, LTImage
from pdfminer.image import ImageWriter
from cStringIO import StringIO
import sys
import os


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

    records = []
    i = 1
    for page in PDFPage.get_pages(fp, pagenos, maxpages=maxpages, password=password,
                                  caching=caching, check_extractable=True):
        # process page
        interpreter.process_page(page)

        # only select lines from the line containing 'Tool' to the line containing "1 The 'All'"
        lines = retstr.getvalue().splitlines()

        idx = containsSubString(lines, 'Tool')
        lines = lines[idx+1:]
        idx = containsSubString(lines, "1 The 'All'")
        lines = lines[:idx]

        for line in lines:
            records.append(line)
        i += 1

    fp.close()
    device.close()
    retstr.close()

    return records


def containsSubString(list, substring):
    # find a substring in a list item
    for i, s in enumerate(list):
        if substring in s:
            return i
    return -1


# process pdf
fn = '../test1.pdf'
ft = 'test.txt'

text = pdfToText(fn)
outFile = open(ft, 'w')
for i in range(0, len(text)):
    outFile.write(text[i])
outFile.close()

它生成一个文本文件并获取所有文本,但是,x没有保留间距。输出如下: enter image description here

文本文档中的x只是一个空格

现在,我只是生成文本输出,但我的目标是生成一个包含表中数据的html文档。我一直在寻找OCR的例子,其中大多数看起来很混乱或不完整。我愿意使用C语言或其他任何可能产生我想要的结果的语言。

编辑:将有多个这样的PDF,我需要从中获取表数据。据我所知,所有PDF的标题都是相同的。


Tags: infromimportforclosepdflinepage
2条回答

尝试Tabula如果它有效,请使用tabula-extractor library(用ruby编写)以编程方式提取数据。

我想起来了,我走错了方向。我所做的是在pdf中创建每个表的png,现在我正在使用opencv&python处理图像。

相关问题 更多 >