python3 获取某个文件夹所有的pdf文件表格提取表格并一起合并到excel文件

下面是一个完整的示例,其中包括了merge_tables_to_excel函数的定义,并且假设该函数的功能是从每个PDF文件中提取第一个表格并将其合并到一个Excel文件中:

python 复制代码
import os  
from pathlib import Path  
import pandas as pd  
import pdfplumber  
  
def extract_first_table_from_pdf(pdf_path):  
    try:  
        with pdfplumber.open(pdf_path) as pdf:  
            for page in pdf.pages:  
                tables = page.extract_tables()  
                if tables:  
                # tables[0]   pdf中的第一个表格,如果pdf有第二个表格你可以修改为tables[1] 根据你需求来调整
                    return tables[0]   
    except Exception as e:  
        print(f"Error reading {pdf_path}: {e}")  
        return None  
  
def merge_tables_to_excel(pdf_files, excel_path):  
    all_tables = []  
    for pdf_path in pdf_files:  
        first_table = extract_first_table_from_pdf(pdf_path)  
        if first_table:  
            df_table = pd.DataFrame(first_table[1:], columns=first_table[0])  
            all_tables.append(df_table)  
      
    if all_tables:  
        merged_tables_df = pd.concat(all_tables, ignore_index=True)  
        merged_tables_df.to_excel(excel_path, sheet_name='Merged Tables', index=False)  
        print(f"Tables have been saved to {excel_path}")  
    else:  
        print("No tables found in the PDF files.")  
  
def find_all_pdf_files(directory):  
    return list(Path(directory).glob("*.pdf"))  
  
if __name__ == "__main__":  
    # 指定PDF文件夹路径  
    pdf_folder = Path("refer")  
    # 获取文件夹中所有的PDF文件  
    pdf_files = find_all_pdf_files(pdf_folder)  
      
    # 打印找到的PDF文件列表  
    print("Found PDF files:", [str(file) for file in pdf_files])  
      
    # 指定要保存的Excel文件路径  
    excel_path = "merged_tables.xlsx"  
      
    # 提取并合并表格数据到Excel文件  
    merge_tables_to_excel(pdf_files, excel_path)

运行结果如图所示:

如果你想要遍历一个文件夹中的所有子文件夹,并获取每个子文件夹中的PDF文件,你可以使用递归函数来实现这个功能。下面是一个修改后的代码示例,它会递归地搜索指定目录及其所有子目录中的PDF文件:

python 复制代码
import os  
from pathlib import Path  
  
def find_all_pdf_files(directory):  
    pdf_files = []  
    for root, dirs, files in os.walk(directory):  
        for file in files:  
            if file.lower().endswith('.pdf'):  
                pdf_files.append(Path(root) / file)  
    return pdf_files  
  
if __name__ == "__main__":  
    # 指定PDF文件夹路径  
    pdf_folder = Path("refer")  
      
    # 获取文件夹中所有的PDF文件,包括子文件夹中的PDF文件  
    pdf_files = find_all_pdf_files(pdf_folder)  
      
    # 打印找到的PDF文件列表  
    print("Found PDF files:", [str(file) for file in pdf_files])  
      
    # 指定要保存的Excel文件路径  
    excel_path = "merged_tables.xlsx"  
      
    # 提取并合并表格数据到Excel文件  
    merge_tables_to_excel(pdf_files, excel_path)

在这个示例中,find_all_pdf_files 函数使用 os.walk() 来递归遍历目录和子目录。os.walk() 会为每个目录返回一个三元组,包含当前目录的路径、当前目录下的子目录名列表,以及当前目录下的文件名列表。函数遍历每个文件名,检查它是否以 .pdf 结尾(不区分大小写),如果是,则将其添加到 pdf_files 列表中。

请确保你的 merge_tables_to_excel 函数能够处理多个PDF文件中的表格合并到Excel文件的逻辑。如果你需要更具体的帮助来定义这个函数,请提供更多关于你希望如何合并表格的信息。

相关推荐
大模型真好玩5 分钟前
深入浅出LangGraph AI Agent智能体开发教程(一)—全面认识LangGraph
人工智能·python·mcp
技术程序猿华锋1 小时前
学术/报告场景实测:从申请OpenAI API Key获取并实现GPT-5 PDF分析机器人(含源码)
gpt·机器人·pdf
花落文心1 小时前
使用 html2canvas + jspdf 实现页面元素下载为pdf文件
前端·javascript·pdf
阿幸软件杂货间1 小时前
PDF补丁丁:开源多年,完全免费的多功能 PDF 工具箱
pdf
FOAF-lambda1 小时前
像WPS Office 一样处理pdf页面尺寸
pdf
乘乘凉2 小时前
Python中函数的闭包和装饰器
前端·数据库·python
爱隐身的官人7 小时前
爬虫基础学习-爬取网页项目(二)
前端·爬虫·python·学习
刘恒1234567899 小时前
Pycharm
ide·python·pycharm
今***b11 小时前
Python 操作 PPT 文件:从新手到高手的实战指南
java·python·powerpoint
程序员杰哥12 小时前
Jmeter+Jenkins接口压力测试持续集成
自动化测试·软件测试·python·测试工具·jmeter·jenkins·压力测试