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

lsdnh521

于 2015-10-01 22:58:00 发布

阅读量2.8k

点赞数

分类专栏：机器学习/大数据文章标签：遍历 python

本文链接：https://blog.youkuaiyun.com/lsdnh521/article/details/48850501

版权

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

13 篇文章

订阅专栏

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)