python获取指定扩展名的文件_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程序设计有所帮助。