Python——查看目录下所有的目录和文件
写程序我们经常会遇到需要遍历某一个目录下的所有文件这个操作，然而python有现成的库，只需要2个循环就可以搞定。
1 import os 2 3 def all_path(dirname): 4 5 result = []#所有的文件 6 7 for maindir, subdir, file_name_list in os.walk(dirname): 8 9 print("1:",maindir) #当前主目录 10 print("2:",subdir) #当前主目录下的所有目录 11 print("3:",file_name_list) #当前主目录下的所有文件 12 13 for filename in file_name_list: 14 apath = os.path.join(maindir, filename)#合并成一个完整路径 15 result.append(apath) 16 17 return result 18 19 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)
