首页 > 代码库 > 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中