-- coding: utf-8 --

import os  
  
def file_name(file_dir):   
    for root, dirs, files in os.walk(file_dir):  
        print(root) #当前目录路径  
        print(dirs) #当前路径下所有子目录  
        print(files) #当前路径下所有非目录子文件

import os
para=r"D:\Caffe-fangxu\caffe-master\examples\twolable\train\n02085936-Maltese_dog"
lable="3"
def file_name(file_dir):   
    for root,dirs,files in os.walk(file_dir):  
        #print(root) #当前目录路径  
       # print(dirs) #当前路径下所有子目录  
        return [files,root]#当前路径下所有非目录子文件
if __name__ == '__main__':
    file=file_name(para)
    f=open(file[1]+"\\test.txt","w")
    for x in file[0]:
        f.write(x+" "+lable+"\n")
    f.close()