python遍历读取文件下的所有指定类型的文件
2018年08月12日 09:53:23 Blue__Bubble 阅读数:1075
#!/usr/bin/env python3
-- coding:utf-8 --
1. for-in dir/subdir to get the filesname
2. splitext filename to filter
import os
def getFiles(dir, suffix): # 查找根目录,文件后缀
res = []
for root, directory, files in os.walk(dir): # =>当前根,根下目录,目录下的文件
for filename in files:
name, suf = os.path.splitext(filename) # =>文件名,文件后缀
if suf == suffix:
res.append(os.path.join(root, filename)) # =>吧一串字符串组合成路径
return res
for file in getFiles("./", '.py'): # =>查找以.py结尾的文件
print(file)