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()

 

相关文章:

  • 2021-08-25
  • 2022-12-23
  • 2022-12-23
  • 2022-12-23
  • 2022-12-23
  • 2022-12-23
  • 2022-12-23
  • 2022-12-23
猜你喜欢
  • 2022-12-23
  • 2022-12-23
  • 2022-01-09
  • 2022-12-23
  • 2022-12-23
  • 2022-12-23
  • 2022-12-23
相关资源
相似解决方案