python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > python获取指定路径下所有指定后缀文件

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

作者:两把刷子

这篇文章主要介绍了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程序设计有所帮助。

您可能感兴趣的文章:
阅读全文