python 把目录下的文件名批量写入txt的简单方法

 

假设 '/home/Data/imgs/' 是你存放批量文件的位置

import os 

trainPath = '/home/Data/imgs/'

for root, dirs, files in os.walk(trainPath): # 遍历这个文件夹下每个文件
    for file in files:
        f = open('train.txt','a') # 没有则创建txt文件，与代码文件同目录
        f.write('/home/Data/imgs/'+file+'.png\n') # png是后缀，根据文件类型修改
        f.close()

最好单独创建一个文件create_list, 以上代码写入这个文件下