#!/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()
【Python】提取指定文件夹中指定类型的所有文件
最新推荐文章于 2023-11-04 14:28:37 发布