获取指定路径下特定后缀的文件

最新推荐文章于 2022-08-09 21:54:13 发布

aoren3824

最新推荐文章于 2022-08-09 21:54:13 发布

阅读量126

点赞数

文章标签：数据结构与算法

原文链接：http://www.cnblogs.com/wyj690/p/5386971.html

版权

# 获取指定路径下所有指定后缀的文件
# dir 指定路径
# ext 指定后缀，链表&不需要带点或者不指定。例子：['xml', 'java']
import os
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

GetFileFromThisRootDir('C:\\data\\1','.txt')

转载于:https://www.cnblogs.com/wyj690/p/5386971.html