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

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文件的場景中。希望本文對你有所幫助，祝你在數據處理的工作中更加得心應手！