读取pdf 表格

最新推荐文章于 2024-04-14 15:15:00 发布

与久i

最新推荐文章于 2024-04-14 15:15:00 发布

阅读量252

点赞数

文章标签： python java servlet

本文链接：https://blog.csdn.net/feng19980817/article/details/127010169

版权

该代码实现了一个将PDF中的表格数据提取并保存到Excel的工作流程。首先读取PDF文件，然后使用pdfplumber库获取PDF的元数据和页面数。接着，遍历每一页，提取表格数据，并通过openpyxl库将数据写入Excel工作簿。最后，保存Excel文件并关闭PDF。此工具适用于处理包含表格信息的PDF文档。

摘要由CSDN通过智能技术生成

import pdfplumber
from openpyxl import Workbook

class PDF(object):
def __init__(self, file_path):
self.pdf_path = file_path
# 读取pdf
try:
self.pdf_info = pdfplumber.open(self.pdf_path)
print('读取文件完成！')
except Exception as e:
print('读取文件失败：', e)

# 打印pdf的基本信息、返回字典，作者、创建时间、修改时间/总页数
def get_pdf(self):
pdf_info = self.pdf_info.metadata
pdf_page = len(self.pdf_info.pages)
print('pdf共%s页' % pdf_page)
print("pdf文件基本信息：\n", pdf_info)
self.close_pdf()

# 提取表格数据,并保存到excel中
def get_table(self):
wb = Workbook() # 实例化一个工作簿对象
ws = wb.active # 获取第一个sheet
con = 0
try:
# 获取每一页的表格中的文字，返回table、row、cell格式：[[[row1],[row2]]]
for page in self.pdf_info.pages:
for table in page.extract_tables():
for row in table:
# 对每个单元格的字符进行简单清洗处理
row_list = [cell.replace('\n', ' ') if cell else '' for cell in row]
ws.append(row_list) # 写入数据
con += 1
print('---------------分割线,第%s页---------------' % con)
except Exception as e:
print('报错：', e)
finally:
wb.save('\\'.join(self.pdf_path.split('\\')[:-1]) + '\pdf_excel.xlsx')
print('写入完成！')
self.close_pdf()

# 关闭文件
def close_pdf(self):
self.pdf_info.close()

if __name__ == "__main__":
file_path = input('请输入pdf文件路径：')
pdf_info = PDF(file_path)
# pdf_info.get_pdf() # 打印pdf基础信息
# 提取pdf表格数据并保存到excel中,文件保存到跟pdf同一文件路径下
pdf_info.get_table()