用Python快速遍历指定路径下的某类文件

import os
def getflist1(path,suffix): #获取每个路径下的suffix后缀的文件

   #用列表推导式实现,简洁,优雅。。。。。。
    res=[os.path.join(path,fname) for fname in os.listdir(path) if fname.endswith(suffix)]
    return res

path=r'E:\BigCSv等1个文件\BigCSv'
res=getflist1(path,'.py')

print(res)

#function getflist1 只遍历根目录

def getflist2(path,suffix):
    res=[]
    for root, dirs, files in os.walk(path):
        for f in files:
            if f.endswith(suffix):
                res.append(os.path.join(root, f))

    return res
path=r'E:\BigCSv等1个文件\BigCSv'
res=getflist2(path,'.py')
print(res)
#function getflist2 遍历根目录及子目录

你可能感兴趣的:(python,深度优先遍历)