如何获取指定文件夹下所有文件名称并写入列表

最近有学员在扣丁学堂Python在线学习官网留言，询问老师如何获取指定文件夹下所有文件名称并写入列表，今天小编给给大家分享一下Python 获取指定文件夹下所有文件名称并写入列表的实例，文章中会有代码列出，需要的小伙伴和小编一起来看一下吧。

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+ '
')
file_object.close()

在做深度学习的时候，会有大量的数据，为了方便，可以直接用命令拷贝到一个文件夹下，代码如下：

import shutil 
import os
import os.path
rootdir = "./mjsynth/mnt/ramdisk/max/90kDICT32px"
#rootdir = "./123"
def Test2(rootDir): 
 for lists in os.listdir(rootDir): 
 #如果找到的是图片，则打印出来
 if lists[-3:]=='jpg':
 print lists
 path = os.path.join(rootDir, lists) 
 shutil.copy(path,"./500")
 continue
 #如果找到的是文件夹，则判断，如果名称小于2则递归
 if int(lists)<501:
 path = os.path.join(rootDir, lists) 
 if os.path.isdir(path): 
 Test2(path) 
Test2(rootdir)