Python实现文件夹深度优先遍历

最新推荐文章于 2024-08-28 12:34:50 发布

原创最新推荐文章于 2024-08-28 12:34:50 发布 · 2.8k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#遍历 #python

机器学习/大数据专栏收录该内容

13 篇文章

订阅专栏

本文介绍了一个使用Python实现的深度优先搜索算法来遍历指定目录及其子目录下的所有文件和子目录。通过递归的方式，该算法能够有效地列出路径下的所有内容，并提供了回调函数来打印每个遇到的目录和文件。

import os

def dfs_myDir(path, printDir = None, printFile = None):
    stack = []
    ret = []
    stack.append(path)
    while len(stack) > 0:
        tmp = stack.pop(len(stack) - 1)
        if(os.path.isdir(tmp)):
            ret.append(tmp)
            for item in os.listdir(tmp):
                stack.append(os.path.join(tmp, item))
            if printDir:
                printDir(tmp)
        elif(os.path.isfile(tmp)):
            ret.append(tmp)
            if printFile:
                printFile(tmp)
    return ret

def printDir(path):
    print "dir: " + path

def printFile(path):
    print "file: " + path

d = dfs_myDir(r'E:\AndroidAppDevelop', printDir, printFile)