查看: 56|回复: 0

扣丁学堂Python培训之如何获取指定文件夹下所有文件名称并写入列表

发表于 2018-6-5 11:48:05

最近有学员在扣丁学堂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+ '\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获取指定文件夹下所有文件名称并写入列表的方法,想要了解更多内容的小伙伴可以登录扣丁学堂官网查看更多内容。扣丁学堂是专业的Python培训机构,不仅有专业的老师和与时俱进的课程体系,还有大量的Python在线视频供学员观看学习哦。


文章转载自:扣丁学堂http://www.codingke.com/


回复

使用道具 举报