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

Python  /  管理员 发布于 7年前   133

如下所示：

import osimport os.pathrootdir = "./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()

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

import shutil import osimport os.pathrootdir = "./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)

以上这篇python 获取指定文件夹下所有文件名称并写入列表的实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持。