目录里所有的 或 随机个数的文件与文件夹名,存储到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()

 

posted @ 2022-01-21 13:38  夕西行  阅读(216)  评论(0编辑  收藏  举报