Python——查看目录下所有的目录和文件

写程序我们经常会遇到需要遍历某一个目录下的所有文件这个操作，然而python有现成的库，只需要2个循环就可以搞定。

import os

def all_path(dirname):

    result = []#所有的文件

    for maindir, subdir, file_name_list in os.walk(dirname):

        print("1:",maindir) #当前主目录
        print("2:",subdir) #当前主目录下的所有目录
        print("3:",file_name_list)  #当前主目录下的所有文件

        for filename in file_name_list:
            apath = os.path.join(maindir, filename)#合并成一个完整路径
            result.append(apath)

    return result

print(all_path("E:myTest"))

程序运行结果如下：

　　['E:\myTest\kuang3.png',  'E:\myTest\txt1.txt',  'E:\myTest\txt2.txt',  'E:\myTest\YYGB1.png',  'E:\myTest\YYGB2.png',  'E:\myTest\test1\txt11.txt',  'E:\myTest\test2\txt21.txt',  'E:\myTest\test2\txt22.txt' ]

当然现在我们是获取了所有的文件，如果我们只想要某一个类型的文件，那么也很简单，写个过滤就好了。所以修改后的代码如下：

import os

filter=[".txt"] #设置过滤后的文件类型 当然可以设置多个类型

def all_path(dirname):

    result = []#所有的文件

    for maindir, subdir, file_name_list in os.walk(dirname):

        # print("1:",maindir) #当前主目录
        # print("2:",subdir) #当前主目录下的所有目录
        # print("3:",file_name_list)  #当前主目录下的所有文件

        for filename in file_name_list:
            apath = os.path.join(maindir, filename)#合并成一个完整路径
            ext = os.path.splitext(apath)[1]  # 获取文件后缀 [0]获取的是除了文件名以外的内容

            if ext in filter:
                result.append(apath)

    return result

print(all_path("E:myTest"))

#运行结果
#['E:\myTest\txt1.txt', 'E:\myTest\txt2.txt', 'E:\myTest\test1\txt11.txt', 'E:\myTest\test2\txt21.txt', 'E:\myTest\test2\txt22.txt']

或者

def list_files(self, path):
        files = []
        list_dir = os.walk(path)
        for maindir, subdir, all_file in list_dir:
            
            for filename in all_file:
                apath = os.path.join(maindir, filename)
                if apath.endswith('.txt'):
                    files.append(apath)