plt批量转txt

plt批量转txt

import os
dir1 = r"G:\Py_test\data\Geolife Trajectories 1.3\Data"  #待读取的文件夹
path_list1 = os.listdir(dir1)

save_dir = r"G:\Py_test\data"
for path in path_list1:
    dir2 = os.path.join(dir1, path)
    dir2_tra = dir2 + '\Trajectory'
    plt_name_list = os.listdir(dir2_tra)
    print(2,plt_name_list)
    for plt_name in plt_name_list:
        plt_path = os.path.join(dir2_tra, plt_name)
        with open(plt_path, 'r') as plt:
            data = plt.readlines()
            data = data[6:]
            print(data)
        f1 =r'G:\Py_test\data\Geolife_Data'
        f2 =r'G:\Py_test\data\Geolife_Data' + dir2[-4:]
        made_dir = os.listdir(f1)
        if dir2[-3:] not in made_dir: #
            os.makedirs(f2)
        save_path = f2 + '\\' + plt_name[:-3] + 'txt'
        print(save_path)
        with open(save_path, 'w') as sp:
            data_lines = ''
            for line in data:
                data_lines += line
            sp.write(data_lines)