数据集txt格式划分为多个txt文件夹

简单记录一下数据标签txt格式划分为多个文件，通常我们标注号的标签，都是在一个txt文件夹中，我们训练的时候需要把txt中的标签按照一定的比例划分为多个文件，这里贴出划分为三个文件的代码，如果你需要划分为两个文件，那么可以将划分好的另外的一个文件复制粘贴到你需要的文件中。代码如果下：

import os
import random
l_train=[]
l_val=[]
l_test=[]

# 读取文件中的内容，并将其打乱写入列表FileNameList
def ReadFileDatas(original_filename):
    file=open(original_filename,'r+')
    FileNameList=file.readlines()
    random.shuffle(FileNameList)
    file.close()
    print("数据集总量：", len(FileNameList))
    return FileNameList
#将数据集随机划分
def TrainValTestFile(FileNameList):
    i=0
    j=len(FileNameList)
    for line in FileNameList:
        if i<(j*0.6):
            i+=1
            l_train.append(line)
        elif i<(j*0.8):
            i+=1
            l_val.append(line)
        else:
            i+=1
            l_test.append(line)
    print("总数量:%d,此时创建train,val,test数据集"%i)
    return l_train,l_val,l_test
#将获取到的各个数据集的包含的文件名写入txt中
def WriteDatasToFile(listInfo, new_filename):
    file_handle = open(new_filename,'w')
    for str_Result in listInfo:
        file_handle.write(str_Result)
    file_handle.close()
    print('写入 %s 文件成功.' % new_filename)
if __name__ == "__main__":
      listFileInfo = ReadFileDatas('D:\\Software\\xx\\tt\\test1.txt') # 读取文件
      l_train,l_val,l_test=TrainValTestFile(listFileInfo)
      WriteDatasToFile(l_train, 'D:\\Software\\xx\\train.txt')
      WriteDatasToFile(l_val, 'D:\\Software\\xx\\val.txt')
      WriteDatasToFile(l_test, 'D:\\Software\\xx\\test.txt')