获取指定路径文件夹的名称，并存入text文本

代码在python3.*环境可以直接运行，注意文件路径要正确，train_list.txt是自己新建的文本文档，当然也可以用程序来创建

# -*- coding: utf-8 -*-
import os
import os.path
rootdir = "D:\Python\*****"
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()