首页 > 代码库 > Python 列取两级目录并保存到list.txt中
Python 列取两级目录并保存到list.txt中
很多时候需要列出某目录下各个子目录中文件列表并保存到list.txt中,以便于以后遍历该子目录
在图片处理中,常用该方法处理图片目录
目录结构分为3级
rootdir
subdir1
pic1
pic2
subdir2
pic1
pic2
想要结果list.txt
rootdir 目录下
list.txt 内容为
subdir1
subdir2
subdir1/2目录下
list.txt内容为
pic1
pic2
代码为:
使用方法,把.py放到要提取的当前目录下即可
import os
rootdir=".";
folderlist=os.listdir(rootdir);
#print (folderlist)
rootlist=open(rootdir+"/list.txt",‘w‘)
for subfolder in folderlist:
#if subfolder[-3:]==".py":
#continue
print (subfolder)
subpath=rootdir+"/"+subfolder
if os.path.exists(subfolder+"/")==False:
print ("NO")
continue
rootlist.write(subfolder+"\n")
#print (subpath)
subfolderlist=os.listdir(subpath)
#print (subfolderlist)
listpath=subpath+"/list.txt"
listtxt=open(listpath,‘w‘)
for line in subfolderlist:
if line[-4:]==".txt":
continue
listtxt.write(line+"\n")
listtxt.close()
rootlist.close()
Python 列取两级目录并保存到list.txt中