遍历文件夹写入txt

将一级目录下，二级目录内的图片名字读成标签数据形式的txt文件

# coding=UTF-8

import os
import sys

path="/home/cdli/Original_10.18"
color_list=os.listdir(path)
l1=len(color_list)-1
i=0
with open("train_color.txt","w") as f:
        for c_l in color_list:
                path_2=path+"/"+c_l+"/"
                pic=os.listdir(path_2)
                print(c_l+','+str(i))
                for p in pic:
                        f.write('/home/cdli/Original_10.18/'+c_l+'/'+p+' '+str(i)+'\n')
                i=i+1