【Python】提取指定文件夹中指定类型的所有文件

zhaotun123

于 2019-08-16 13:42:52 发布

阅读量2.2k

点赞数 2

分类专栏： python 文章标签： python

本文链接：https://blog.youkuaiyun.com/zhaotun123/article/details/99676405

版权

python 专栏收录该内容

6 篇文章

订阅专栏

本文介绍了一种使用Python批量读取指定目录下特定类型文件的方法，并将其路径写入txt文件中，适用于需要批量处理文件的场景。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#!/usr/bin/python

import os


def ListFilesToTxt(dir,file,wildcard,recursion):
    exts = wildcard.split(" ")
    files = os.listdir(dir)
    for name in files:
        fullname=os.path.join(dir,name)
        if(os.path.isdir(fullname) & recursion):
            ListFilesToTxt(fullname,file,wildcard,recursion)
        else:
            for ext in exts:
                if(name.endswith(ext)):
                    file.write(fullname + "\n")
                    break

def Test():
  dir = "train-1118-24/"                       #文件夹路径，提取该文件夹中所有文件的路径
  outfile = "train-1118-24.txt"                #写入的txt文件名
  wildcard = ".jpg .txt .exe .dll .lib"      #要读取的文件类型；

  file = open(outfile,"w")
  if not file:
    print ("cannot open the file %s for writing" % outfile)

  ListFilesToTxt(dir,file,wildcard, 1)

  file.close()


Test()