获取文件夹下的文件名并存入txt中

import os path_imgs='C:/Users/13451/Desktop/pic' #图片存放的地址 for files in os.listdir(path_imgs): #listdir函数获取文件夹下图片的名字,返回包含所有文件名字的列表 with open('C:/Users/13451/Desktop/train_data.txt','a') as f: f.write(files.rstrip('.png')) #只保存名字去除后缀.png f.write("\n")