Python文件移动

import shutil
import os
from utils import train_test_split

def remove_file(old_path, new_path):
    print(old_path)
    print(new_path)
    filelist = os.listdir(old_path)  # 列出该目录下的所有文件,listdir返回的文件列表是不包含路径的。
    print(filelist)
    for file in filelist:
        src = os.path.join(old_path, file)
        dst = os.path.join(new_path, file)
        print('src:', src)
        print('dst:', dst)
        shutil.move(src, dst)


if __name__ == '__main__':
    #将roi移动到数据集中
    remove_file(r"D:\PycharmProjects\ShowPalmIdentification\Palm_Vein_Identification\addroi", r"D:\PycharmProjects\ShowPalmIdentification\Palm_Vein_Dataset\palm_vein_dataset_size128")
    #更新txt中数据
    txtfile = open('D:\PycharmProjects\ShowPalmIdentification\Palm_Vein_Identification\ZhangkeTeam.txt', 'w')
    for file in os.listdir(r"D:\PycharmProjects\ShowPalmIdentification\Palm_Vein_Dataset\palm_vein_dataset_size128"):
        txtfile.write(file +" "+str(file[0])+'\n')
    #txtfile.close()

    train_test_split("D:\PycharmProjects\ShowPalmIdentification\Palm_Vein_Identification\ZhangkeTeam.txt","ZhangkeTeam")

def generate_txt(start, end, file):
    """
    按照要求生成txt文件
    :param start: 起始
    :param end: 结束
    :param file: 文件
    :return:
    """
    with open(file, 'a+') as f:
        for i in range(start, end+1):
            for j in range(1, 3):
                for k in range(1, 26):
                    f.write('{}_{}_2_{}.png\t{}\n'.format(i, j, k, i-1))