Python for NLP：如何处理包含多个标题和子标题的PDF文本？

import PyPDF2

# 打开PDF文件
pdf_file = open('example.pdf', 'rb')

# 创建一个PDF读取对象
pdf_reader = PyPDF2.PdfFileReader(pdf_file)

# 获取PDF中的页数
num_pages = pdf_reader.numPages

# 逐页读取文本
text = []
for page_num in range(num_pages):
    page = pdf_reader.getPage(page_num)
    text.append(page.extract_text())

# 关闭PDF文件
pdf_file.close()

import re

# 定义标题和子标题的正则表达式
title_pattern = r'^d+.s(.+)$'  # 例如：1. 标题
sub_title_pattern = r'^d+.d+.s(.+)$'  # 例如：1.1. 子标题

# 提取标题和子标题
titles = []
sub_titles = []

for page in text:
    lines = page.split('
')
    for line in lines:
        title_match = re.match(title_pattern, line)
        sub_title_match = re.match(sub_title_pattern, line)
        
        if title_match:
            title = title_match.group(1)
            titles.append(title)
        elif sub_title_match:
            sub_title = sub_title_match.group(1)
            sub_titles.append(sub_title)