python 之遍历目录树（可匹配输出特定后缀的文件）

　　涉及到的模块有os, fnmatch：1、通过os模块中的方法获取dir、subdir、files，通过os.path.join可拼接成完整路径；

　　　　　　　　　　　　　　　　2、fnmatch主要通过fnmatch.fnmatch(name, patterns)，在patterns中匹配name元素，用于获取特定后缀的文件。

　　可将这一功能代码封装，以便后续调用：

　　

 1 #!/usr/bin/env python
 2 import os, fnmatch
 3 
 4 def all_files(root, patterns='*', single_level=False, yield_folders=False):
 5     #将模式从字符串中取出放入列表中
 6     patterns = patterns.split(';') #可以指定多个后缀作为需要获取的文件类型，通过';'作为分隔符
 7     for path, subdir, files in os.walk(root):
 8         if yield_folders :
 9             files.extend(subdir)
10         files.sort()
11         for name in files:
12             for pattern in patterns:
13                 if fnmatch.fnmatch(name, patterns):
14                     yield os.path.join(path, name)
15                     break
16     if single_level:
17         break:

    为了验证最终结果，（注意，最终结果是可迭代的），所以本人具体实例验证如下：

　　

1 for path in all_files('/usr/local/test', pattern='*.py'):
2     print path

　　。

　　与大家共勉！

转载于:https://www.cnblogs.com/Tony-zhangl/p/4238818.html