Python遍历文件夹的两种方法比较

Python遍历文件夹的两种方法比较

转自：点击打开链接

遍历文件夹是一个很常用的功能吧。这里分别用两种方法实现： 

第一种：使用os.walk:

# -*- coding: utf-8 -*- import os def Test1(rootDir):     list_dirs = os.walk(rootDir)     for root, dirs, files in list_dirs:         for d in dirs:             print os.path.join(root, d)              for f in files:             print os.path.join(root, f)

第二种：使用os.listdir:

# -*- coding: utf-8 -*- import os def Test2(rootDir):     for lists in os.listdir(rootDir):         path = os.path.join(rootDir, lists)         print path         if os.path.isdir(path):             Test2(path) 

这两种到底有什么区别呢？

这里先建立一个测试目录E:\test，目录结构如下：

E:\TEST │--A │  │--A-A │  │  │--A-A-A.txt │  │--A-B.txt │  │--A-C │  │  │--A-B-A.txt │  │--A-D.txt │--B.txt │--C │  │--C-A.txt │  │--C-B.txt │--D.txt │--E 

下面通过运行如下代码：

Test1('E:\TEST') print '=======================================' Test2('E:\TEST') 

输出结果为：

>>>  E:\TEST\A E:\TEST\C E:\TEST\E E:\TEST\B.txt E:\TEST\D.txt E:\TEST\A\A-A E:\TEST\A\A-C E:\TEST\A\A-B.txt E:\TEST\A\A-D.txt E:\TEST\A\A-A\A-A-A.txt E:\TEST\A\A-C\A-B-A.txt E:\TEST\C\C-A.txt E:\TEST\C\C-B.txt ======================================= E:\TEST\A E:\TEST\A\A-A E:\TEST\A\A-A\A-A-A.txt E:\TEST\A\A-B.txt E:\TEST\A\A-C E:\TEST\A\A-C\A-B-A.txt E:\TEST\A\A-D.txt E:\TEST\B.txt E:\TEST\C E:\TEST\C\C-A.txt E:\TEST\C\C-B.txt E:\TEST\D.txt E:\TEST\E >>>  

可以看出，对于第一种方法，输出总是先文件夹后文件名的，对于第二种，则是按照目录树结构以及按照首字母排序进行输出的。

补充加入：对某目录下（包括子目录）的所有.ppm文件改为.jpg文件。

from PIL import Imageimport osimport os.pathdef change_format(rootDir):for root, dirs, files in os.walk(rootDir):for file in files:if os.path.splitext(file)[1] == '.ppm':im = Image.open(os.path.join(root, file))filename = os.path.join(root, file)im.save(os.path.splitext(filename)[0] + '.jpg', 'JPEG')if __name__ == '__main__':_dir = "your directory"change_format(_dir)

root 为 path（一直到file名字（除去文件名字）），file只是文件名字加上扩展名
此处os.path.join(root,file)为absolute path，os.path.splitext(file)[1]为扩展名（os.path.splitext(file)[1][1:]为ppm不带.)，os.path.splitext(file)[0]为文件名