Python+os+openpyxl怎么批量获取Excel的文件名和最大行数

在日常的数据处理工作中，我们经常需要处理大量的Excel文件。为了高效地管理这些文件，我们可能需要批量获取这些Excel文件的文件名以及每个文件的最大行数。本文将介绍如何使用Python的os模块和openpyxl库来实现这一目标。

1. 准备工作

在开始之前，我们需要确保已经安装了openpyxl库。如果还没有安装，可以使用以下命令进行安装：

pip install openpyxl

2. 导入必要的模块

首先，我们需要导入os模块和openpyxl库。os模块用于处理文件和目录路径，而openpyxl库则用于读取Excel文件。

import os
from openpyxl import load_workbook

3. 获取指定目录下的所有Excel文件

假设我们有一个目录，里面存放了多个Excel文件。我们可以使用os.listdir()函数来获取该目录下的所有文件名，然后筛选出Excel文件。

def get_excel_files(directory):
    # 获取目录下的所有文件
    files = os.listdir(directory)
    
    # 筛选出Excel文件
    excel_files = [file for file in files if file.endswith('.xlsx') or file.endswith('.xls')]
    
    return excel_files

4. 获取Excel文件的最大行数

接下来，我们需要编写一个函数来获取每个Excel文件的最大行数。我们可以使用openpyxl库中的load_workbook()函数来加载Excel文件，然后通过sheet.max_row属性获取最大行数。

def get_max_row(file_path):
    # 加载Excel文件
    wb = load_workbook(file_path)
    
    # 获取第一个工作表
    sheet = wb.active
    
    # 返回最大行数
    return sheet.max_row

5. 批量获取文件名和最大行数

现在，我们可以将上述两个函数结合起来，批量获取指定目录下所有Excel文件的文件名和最大行数。

def batch_get_file_info(directory):
    # 获取目录下的所有Excel文件
    excel_files = get_excel_files(directory)
    
    # 存储文件名和最大行数的列表
    file_info = []
    
    # 遍历每个Excel文件
    for file in excel_files:
        # 获取文件路径
        file_path = os.path.join(directory, file)
        
        # 获取最大行数
        max_row = get_max_row(file_path)
        
        # 将文件名和最大行数添加到列表中
        file_info.append((file, max_row))
    
    return file_info

6. 示例代码

下面是一个完整的示例代码，展示了如何使用上述函数来批量获取Excel文件的文件名和最大行数。

import os
from openpyxl import load_workbook

def get_excel_files(directory):
    files = os.listdir(directory)
    excel_files = [file for file in files if file.endswith('.xlsx') or file.endswith('.xls')]
    return excel_files

def get_max_row(file_path):
    wb = load_workbook(file_path)
    sheet = wb.active
    return sheet.max_row

def batch_get_file_info(directory):
    excel_files = get_excel_files(directory)
    file_info = []
    for file in excel_files:
        file_path = os.path.join(directory, file)
        max_row = get_max_row(file_path)
        file_info.append((file, max_row))
    return file_info

if __name__ == "__main__":
    # 指定目录
    directory = "path/to/your/excel/files"
    
    # 获取文件名和最大行数
    file_info = batch_get_file_info(directory)
    
    # 打印结果
    for file, max_row in file_info:
        print(f"文件名: {file}, 最大行数: {max_row}")

7. 运行结果

假设我们有一个目录/data/excel_files，里面存放了三个Excel文件：file1.xlsx、file2.xlsx和file3.xlsx。运行上述代码后，输出结果可能如下：

文件名: file1.xlsx, 最大行数: 100
文件名: file2.xlsx, 最大行数: 200
文件名: file3.xlsx, 最大行数: 150

8. 总结

通过本文的介绍，我们学习了如何使用Python的os模块和openpyxl库来批量获取Excel文件的文件名和最大行数。这种方法不仅简单高效，而且可以轻松扩展到处理大量Excel文件的场景中。希望本文对你有所帮助，祝你在数据处理的工作中更加得心应手！