目录里所有的 或 随机个数的文件与文件夹名,存储到txt中
1、目录里所有的文件与文件夹名,存储到txt中
import os fileNames=os.listdir('C:/Users/Administrator/Desktop/dataset-master/Annotations') #目录里所有文件、文件夹名 txt_save=open('D:/result.txt','w') #待写入的txt for fileName in fileNames: fileName=fileName[:-4]+'\n' #只取文件名,不带后缀(.xml) txt_save.write(fileName) txt_save.close()
2、目录里随机个文件与文件夹名,存储到txt中
import os import random fileNames=os.listdir('C:/Users/Administrator/Desktop/dataset-master/Annotations') #目录里所有文件、文件夹名 txt_save=open('D:/result.txt','w') #待写入的txt totality=len(fileNames) #总数 lists=range(totality) #0,1,2,……数列 selects_lists=random.sample(lists,int(0.2*totality)) #数列里随机取20%个元素(随机无序的) for select in selects_lists: fileName=fileNames[select][:-4]+'\n' #只取文件名,不带后缀(.xml) txt_save.write(fileName) txt_save.close()