Python遍历目录2021-08-27

递归遍历目录

import os
def getAllDirRe(path, sp = ""):
    #得到当前目录下的所有文件
    filesList = os.listdir(path)
    sp += "   "
    #处理每个文件
    for fileName in filesList:
        #判断是否为路径
        fileAbsPath = os.path.join(path, fileName)
        if os.path.isdir(fileAbsPath):
            print(sp + "目录：" + fileName)
            #递归调用
            getAllDirRe(fileAbsPath, sp)
        else:
            print(sp + "普通文件：" + fileName)
path = r"D:\My Documents"
getAllDirRe(path)

栈模拟递归遍历目录（深度遍历）

import os
def getAllDirDE(path):
    stack = []
    stack.append(path)
    #处理栈，当栈为空的时候结束循环
    while len(stack) != 0:
        #从栈里取出数据
        dirPath = stack.pop()
        #目录下所有文件
        filesList = os.listdir(dirPath)
        #处理每个文件，如果是普通文件则打印出来，如果是目录，则压栈
        for fileName in filesList:
            fileAbsPath = os.path.join(dirPath, fileName)
            if os.path.isdir(fileAbsPath):
                #是目录，压栈
                print("目录：" + fileName)
                stack.append(fileAbsPath)
            else:
                #打印普通文件
                print("普通文件：" + fileName)
path = r"D:\My Documents"
getAllDirDE(path)

队列模拟递归遍历目录（广度遍历）

import os
from collections import deque
def getAllDirQU(path):
    queue = deque()
    #进队
    queue.append(path)
    while len(queue) != 0:
        #出队数据
        dirPath = queue.popleft()
        #找出所有的文件
        filesList = os.listdir(dirPath)
        for fileName in filesList:
            #绝对路径
            fileAbsPath = os.path.join(dirPath, fileName)
            #判断是否是目录，是目录进队，不是就打印
            if os.path.isdir(fileAbsPath):
                print("目录：" + fileName)
                queue.append(fileAbsPath)
            else:
                print("普通文件：" + fileName)
path = r"D:\My Documents"
getAllDirQU(path)