目录
- 1、PyPDF2
- 2、pdfplumber
- 3、PyMuPDF
- 4、pdfminer
- 总结
1、PyPDF2
PyPDF2
是一个常用的库,可以用来读取、合并、分割和修改PDF文件。读取pdf内容:
import PyPDF2
# 打开PDF文件
with open('ELLK-Net_An_Efficient_Lightweight_Large_Kernel_Network_for_SAR_Ship_Detection.pdf', 'rb') as file:
reader = PyPDF2.PdfReader(file)
# 获取页面数
num_pages = len(reader.pages)
print(f"页面数: {num_pages}")
# 读取每一页的文本
for i in range(num_pages):
page = reader.pages[i]
text = page.extract_text()
print(f"第 {i + 1} 页的内容:\n{text}")
结果如下:
效果: 可以看出,读取是从左往右,从上往下,文本内容读取的还挺全的,但是图片、表格都读不出来。
2、pdfplumber
pdfplumber
是一个功能强大的库,专门用于提取PDF文件中的文本和表格数据。读取PDF内容:
import pdfplumber
# 打开PDF文件
with pdfplumber.open('example.pdf') as pdf:
for i in range(len(pdf.pages)):
page = pdf.pages[i]
text = page.extract_text()
print(f"第 {i + 1} 页的内容:\n{text}")
结果如下:
效果: 看结果,这个库读取PDF是一整行一整行读取的,当文档内容是两页的时候,两列同一行会被同时读取。看别人说读取表格方便,但是这里也没读取出来,具体如下图所示。
此外,使用table = page.extract_table()
单独读取有表格的页,输出结果为None
3、PyMuPDF
PyMuPDF
是另一个高效的库,可以处理PDF文件,支持文本提取、图像提取等功能。读取pdf如下:
import fitz # PyMuPDF
from PIL import Image
import io
# 打开PDF文件
doc = fitz.open('ELLK-Net_An_Efficient_Lightweight_Large_Kernel_Network_for_SAR_Ship_Detection.pdf')
for page_number,page in enumerate(doc):
text = page.get_text()
print(f"第 {page.number + 1} 页的内容:\n{text}")
# 获取页面中的所有图片
image_list = page.get_images(full=True)
# 遍历所有图片
for img_index, img in enumerate(image_list):
# 图片的XREF
xref = img[0]
# 获取图片对象
base_image = doc.extract_image(xref)
# 获取图片的二进制数据
image_bytes = base_image["image"]
# 将二进制数据转为PIL图像
image = Image.open(io.BytesIO(image_bytes))
# # 左右翻转
# image = image.transpose(method=Image.FLIP_LEFT_RIGHT)
# 上下翻转
image = image.transpose(method=Image.FLIP_TOP_BOTTOM)
# 获取图片的扩展名
image_ext = base_image["ext"]
# 保存图片
# with open(f"page_{page_number + 1}_img_{img_index + 1}.{image_ext}", "wb") as image_file:
# image_file.write(image_bytes)
image.save(f"page_{page_number + 1}_img_{img_index + 1}.{image_ext}")
print(f"第 {page_number + 1} 页中提取的图片 {img_index + 1} 已保存。")
读取效果如下:
效果:文字提取效果与PyPDF2差不多,不同点是遇到大的空格会直接换行。可以读取图片,我这里发现读取的图片需要上下翻转。插入的图片可以直接读取,但是流程图可能因为是visio画的,读取不出来。另外,三线表格还是没有读取出来。
4、pdfminer
pdfminer.six
是一个专注于从PDF文件中提取文本和元数据的库。读取PDF如下:
from pdfminer.high_level import extract_text
# 提取文本
text = extract_text('example.pdf')
print(text)
读取结果:
效果: 读取结果整体空行多。读取方式按行读取,有左右两列的时候,哪一列先读到就读哪一列,因此顺序可能会错乱,就像上面结果图一项,该页内容先读取的右边的列。
总结
看网上的说法是:
PyPDF2:适合基本的PDF操作,如合并和提取文本。
pdfplumber:非常适合提取文本和表格。
PyMuPDF:功能强大,处理速度快,支持多种操作。
pdfminer.six:更适合复杂的文本提取需求。
但在读取类似上面这样的论文文档时,PyPDF2和PyMuPDF感觉好用点。