从多份 PDF 格式的简历中提取信息

Question

我有一个数据集，其中有一列包含用于简历的谷歌驱动器链接，我有 5000 行，因此有 5000 个链接，我试图从这些简历中的 2 个单独的列中提取诸如多年经验和工资之类的信息。到目前为止，我已经看到了这里提到的很多例子。

例如：下面提到的代码只能从一个文件读取数据，如何将其复制到多行？

请帮我解决这个问题，否则我将不得不手动浏览 500 份简历并填写数据

，希望我能找到解决这个痛苦问题的方法。

pdf_file = open('sample.pdf', 'rb')
read_pdf = PyPDF2.PdfFileReader(pdf_file)
number_of_pages = read_pdf.getNumPages()
page = read_pdf.getPage(0)
page_content = page.extractText()
print page_content.encode('utf-8')

#to extract salary , experience using regular expressions
import re

prog = re.compile("\s*(Name|name|nick).*")
result = prog.match("Name: Bob Exampleson")

if result:
    print result.group(0)

result = prog.match("University: MIT")

if result:
    print result.group(0)

Answer 1

使用循环。基本上，您将主要代码放入函数中（更易于阅读）并创建文件名列表。然后，您迭代此列表，使用列表中的值作为函数的参数：

注意：我没有检查您的抓取代码，只是展示了如何循环。还有更有效的方法可以做到这一点，但我假设您是 Python 初学者，所以让我们从简单的角度开始吧。

# add your imports to the top
import re

# create a list of your filenames
files_list = ['a.pdf', 'b.pdf', 'c.pdf']
for filename in files_list:  # iterate over the list
    get_data(filename)


# put the rest in a function for readability
def get_data(filename):
    pdf_file = open(filename, 'rb')
    read_pdf = PyPDF2.PdfFileReader(pdf_file)
    number_of_pages = read_pdf.getNumPages()
    page = read_pdf.getPage(0)
    page_content = page.extractText()
    print page_content.encode('utf-8')

    prog = re.compile("\s*(Name|name|nick).*")
    result = prog.match("Name: Bob Exampleson")

    if result:
        print result.group(0)

    result = prog.match("University: MIT")

    if result:
        print result.group(0)

现在您的下一个问题可能是，如何创建包含 5000 个文件名的列表？这取决于文件的名称及其存储位置。如果它们是连续的，您可以执行以下操作：

files_list = []  # empty list
num_files = 5000  # total number of files
for i in range(1, num_files+1):
    files_list.append(f'myfile-{i}.pdf')

这将创建一个包含“myfile-1.pdf”、“myfile-2.pdf”等的列表。

希望这足以让您开始。

您还可以在函数中使用 return 来创建一个新列表，其中包含稍后可以使用的所有输出，而不是随时打印输出：

output = []

def doSomething(i):
    return i * 2

for i in range(1, 100):
    output.append(doSomething(i))

# output is now a list with values like:
# [2, 4, 6, 8, 10, 12, ...]