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 遍历根目录及子目录