数据处理（txt的提取）

本人用于记录平时用的代码，仅供新手参考！！

通常对数据进行处理，如提取txt文件里面的内容，将其分成训练集和测试集，但是测试集需要根据已有的文件名进行挑选，而训练集和测试集也就分开了。

首先，将测试数据里面的文件名进行提取，然后根据该文件同已有的txt进行对比，再对其保存输出

即代码如下：

import sys
import json
import os
fp = open("ttl.txt", "r")
sample = fp.readlines()

path_imgs = './valt/'
for files in os.listdir(path_imgs):
    img_path = files
    with open("valtt.txt", "a") as f:
        f.write(str(img_path) + ' ')
with open('valtt.txt','r',encoding='utf-8') as jaf:
    data = jaf.readline() #读取数据行
    james = data.strip().split(' ')#将数据转换为列表
    #print(james)
with open("valt.txt","w",encoding="utf-8") as f_w:
    for line in sample:
        line = line.strip()
        for i in james:
            if i in line:
               # print(line)
                f_w.write(line)
                f_w.write('\n')

fp.close()