import os



path = "train_data2/train_data2"
path_list = []
for root, dirs, files in os.walk(path):
    for file in files:
        path = os.path.join(root, file)
        path_list.append(path)


data_new = []
for i in path_list:
    with open(i, encoding="utf-8") as f:
        data_str = f.read()
    data_list = data_str.split("\n")
    data_new.append(data_list[0].split("@@@@@")[0])

with open("data/题目5.txt", "w", encoding="utf-8") as f:
    for i in data_new:
        f.write(i)
        f.write("\n")