python 遍历文件夹

一. 用递归遍历目录

# 定义一个函数来遍历目录
import os
def getAllDirAndFile(pathDir):
    # 判断是否是目录
    if not os.path.isdir(pathDir):
        return
    listNames = os.listdir(pathDir)
    for name in listNames:
        # 拼接成绝对路径
        absPath = os.path.join(pathDir,name)
        # 判断是否是目录
        if os.path.isdir(absPath):
            print("目录:%s"%(absPath))
            # 调用自己
            getAllDirAndFile(absPath)
        if os.path.isfile(absPath):
            print("文件:%s"%(absPath))

# 测试
getAllDirAndFile(path)

二. 栈实现深度遍历

import os
import collections
def getAllDirAndFile(pathDir):
    if not os.path.isdir(pathDir):
        return
    #创建一个栈
    strack = collections.deque()

    #把根路径放到队列中去
    strack.append(pathDir)

    while True:
        # 如果队列的长度为0, 则队列中已经没有数据.  不在遍历了
        if len(strack) == 0:
            break

        #把路径取出来遍历
        path = strack.pop()

        listNames = os.listdir(path)
        for name in listNames:
            absPath = os.path.join(path,name)
            if os.path.isfile(absPath):#文件
                print("文件:%s"%(absPath))
            if os.path.isdir(absPath):#目录
                print("目录:%s"%(absPath))
    #           将目录放入到队列中去
                strack.append(absPath)

# 测试:
path = r"C:\Users\Administrator\Desktop"
getAllDirAndFile(path)

3. 队列实现广度遍历

import  os
import collections
def getAllDirAndFile(pathDir):
    if not os.path.isdir(pathDir):
        return
    # 创建一个队列
    queue = collections.deque()

    # 把根路径放到队列中去
    queue.append(pathDir)

    while True:
        if len(queue) == 0:
            break

        # 把路径取出来遍历
        path = queue.popleft()

        listNames = os.listdir(path)
        for name in listNames:
            absPath = os.path.join(path, name)
            if os.path.isfile(absPath):
                print("文件:%s" % (absPath))
            if os.path.isdir(absPath):
                print("目录:%s" % (absPath))
                queue.append(absPath)

# 测试
getAllDirAndFile(r"F:\Program Files")