摘要：統(tǒng)計(jì)指定文件夾下所有的文件數(shù)量本來(lái)一直是有這個(gè)需求，只是以前寫(xiě)的是遞歸的方式處理，感覺(jué)對(duì)資源的占用不友好，而且的最大遞歸深度不超過(guò)，所以改了一下，這里用廣度優(yōu)先遍歷的方式實(shí)現(xiàn)。實(shí)測(cè)兩層共個(gè)文件夾，共的文件數(shù)量。

本來(lái)一直是有這個(gè)需求，只是以前寫(xiě)的是遞歸的方式處理，感覺(jué)對(duì)資源的占用不友好，而且python的最大遞歸深度不超過(guò)1000，所以改了一下，這里用廣度優(yōu)先遍歷的方式實(shí)現(xiàn)。
實(shí)測(cè)兩層共24個(gè)文件夾，共50w的文件數(shù)量。運(yùn)行時(shí)間大概3秒。以下是代碼：

import os
import queue

def get_file_quantity(folder: str) -> int:
    """BFS獲取文件夾下文件的總數(shù)量"""
    # 判斷初始文件夾
    assert os.path.isdir(folder), "請(qǐng)輸入有效的文件夾參數(shù)"
    file_quantity = 0                       # 初始化文件數(shù)量
    folder_path_queue = queue.Queue()
    folder_path_queue.put_nowait(folder)    # 初始化隊(duì)列的值
    # 處理隊(duì)列里的文件夾
    while not folder_path_queue.empty():
        folder = folder_path_queue.get_nowait()
        file_folder_list = list(map(lambda bar: os.path.join(folder, bar), os.listdir(folder)))
        folder_list = list(filter(lambda bar: os.path.isdir(bar), file_folder_list))
        for folder_path in folder_list:
            folder_path_queue.put_nowait(folder_path)
        temp_file_count = len(file_folder_list) - len(folder_list)
        file_quantity += temp_file_count
    return file_quantity
    
if __name__ == "__main__":
    file_quantity = get_file_quantity(r"/home")
    print(f"文件總數(shù)是: {file_quantity}")

這里主要是使用了隊(duì)列，就是普通的BFS的思路

這里稍微改一下folder_list = list(filter(lambda bar: os.path.isdir(bar), file_folder_list))里的的lambda函數(shù)就可以實(shí)現(xiàn)判斷對(duì)文件名的各種判斷操作，這里函數(shù)功能的實(shí)現(xiàn)就完全取決于自己的腦洞了！

附上一個(gè)改編版： 查看包含特定后綴的文件的數(shù)量

import os
import queue

def filter_extension(filename: str, extension: str) -> bool:
    """判斷文件路徑名的后綴是否和給定的后綴字符串相同
    只是單純的字符串判斷
    """
    basename_and_extension = filename.split(".")
    return (basename_and_extension[-1] == extension) and (len(basename_and_extension) >= 2)

def get_file_quantity(folder: str, extension: str) -> int:
    """BFS獲取文件夾下文件的總數(shù)量"""
    # 判斷初始文件夾
    assert os.path.isdir(folder), "請(qǐng)輸入有效的文件夾參數(shù)"
    assert isinstance(extension, str), "請(qǐng)輸入有效的文件后綴名"
    file_quantity = 0                       # 初始化文件數(shù)量
    folder_path_queue = queue.Queue()
    folder_path_queue.put_nowait(folder)    # 初始化隊(duì)列的值
    # 處理隊(duì)列里的文件夾
    while not folder_path_queue.empty():
        folder = folder_path_queue.get_nowait()
        file_folder_list = list(map(lambda bar: os.path.join(folder, bar), os.listdir(folder)))
        folder_list = list(filter(lambda bar: os.path.isdir(bar), file_folder_list))
        file_list = list(filter(lambda bar: os.path.isfile(bar), file_folder_list))
        match_extension_list = list(filter(lambda bar: filter_extension(bar, extension), file_list))
        for folder_path in folder_list:
            folder_path_queue.put_nowait(folder_path)
        temp_file_count = len(match_extension_list)
        file_quantity += temp_file_count
    return file_quantity
if __name__ == "__main__":
    extension = "py"
    file_quantity = get_file_quantity(r"/home", extension)
    print(f"包含后綴 {extension } 的文件的數(shù)量: {file_quantity}")