300字范文,内容丰富有趣,生活中的好帮手!
300字范文 > python 之遍历目录树(可匹配输出特定后缀的文件)

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

时间:2022-05-01 22:01:42

相关推荐

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#将模式从字符串中取出放入列表中 6patterns = patterns.split(';') #可以指定多个后缀作为需要获取的文件类型,通过';'作为分隔符 7for 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 break16if single_level:17 break:

为了验证最终结果,(注意,最终结果是可迭代的),所以本人具体实例验证如下:

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

与大家共勉!

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。