python 获取指定文件夹下所有文件名称并写入列表的实例

import os

import os.path

rootdir = "./pic_data"

file_object = open('train_list.txt','w')

for parent,dirnames,filenames in os.walk(rootdir):

 for filename in filenames:

 print filename

  file_object.write(filename+ '\n')

file_object.close()