码迷,mamicode.com
首页 > 编程语言 > 详细

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

时间:2015-01-21 14:51:50      阅读:180      评论:0      收藏:0      [点我收藏+]

标签:

  涉及到的模块有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

  技术分享

  与大家共勉!

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

标签:

原文地址:http://www.cnblogs.com/Tony-zhangl/p/4238818.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!