Python遍历指定文件夹获取文件列表并写入文件的实现方法

叶初

发布时间：2026-10-13| 7人浏览过

原创

本文介绍在Python中如何遍历指定文件夹以获取所有文件名称，并将结果写入列表或文本文件。通过os.walk和os.listdir两种方法，分别实现递归遍历子目录和非递归获取指定类型文件的功能，适用于深度学习等需要批量处理数据的场景。

具体实现代码如下所示：

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获取指定文件夹下所有文件名称并写入列表的完整示例。希望此内容能为您提供参考，同时也欢迎大家继续支持本站获取更多技术分享。