当前位置: 移动技术网 > IT编程>脚本编程>Python > python 获取指定文件夹下所有文件名称并写入列表的实例

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

2018年04月30日  | 移动技术网IT编程  | 我要评论

3u8831航班,乐巢汇论坛,余洒度

如下所示:

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()

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

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)

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

如对本文有疑问,请在下面进行留言讨论,广大热心网友会与你互动!! 点击进行留言回复

相关文章:

验证码:
移动技术网