Python使用广度优先和深度优先两种方法遍历目录树

代码不长，并且加了很多注释，就不再多说明了。

from os import listdir

from os.path import join, isfile, isdir

def listDirWidthFirst(directory):

    '''广度优先遍历文件夹'''

    #使用列表模拟双端队列，效率稍微受影响，不过关系不大

    dirs = [directory]

    #如果还有没遍历过的文件夹，继续循环

    while dirs:

        #遍历还没遍历过的第一项

        current = dirs.pop(0)

        #遍历该文件夹

        #如果是文件就直接输出显示

        #如果是文件夹，输出显示后，标记为待遍历项

        for subPath in listdir(current):

            path = join(current, subPath)

            if isfile(path):

                print(path)

            elif isdir(path):

                print(path)

                dirs.append(path)

def listDirDepthFirst(directory):

    '''深度优先遍历文件夹'''

    #遍历文件夹

    #如果是文件就直接输出

    #如果是文件夹，就输出显示，然后递归遍历该文件夹

    for subPath in listdir(directory):

        path = join(directory, subPath)

        if isfile(path):

            print(path)

        elif isdir(path):

            print(path)

            listDirDepthFirst(path)

directory = r'F:\test'

listDirWidthFirst(directory)

print('='*30)

listDirDepthFirst(directory)

由于手机屏幕大小不同，很难有统一的排版方式，考虑了一下，对于较短的代码，我把截图一起发上来，这样比较好对照，尤其是代码的缩进。

代码运行结果类似于下面图中的样子：