python获取指定路径下所有指定后缀文件的方法

本文实例讲述了python获取指定路径下所有指定后缀文件的方法。分享给大家供大家参考。具体实现方法如下:

# 获取指定路径下所有指定后缀的文件
# dir 指定路径
# ext 指定后缀,链表&不需要带点 或者不指定。例子:[‘xml’, ‘java’]
def getfilefromthisrootdir(dir,ext = none):
allfiles = []
needextfilter = (ext != none)
for root,dirs,files in os.walk(dir):
for filespath in files:
filepath = os.path.join(root, filespath)
extension = os.path.splitext(filepath)[1][1:]
if needextfilter and extension in ext:
allfiles.append(filepath)
elif not needextfilter:
allfiles.append(filepath)
return allfiles

希望本文所述对大家的python程序设计有所帮助。

Posted in 未分类