python随机移动文件

在处理好数据之后，需要随机划分一些数据作为验证集

# 随机移动1000个文件作为验证集
import os
import random
import shutil

if __name__ == '__main__':
    source_image_dir = "train_nii_06/JPEGImages" # 包含所有的数据
    source_label_dir = "train_nii_06/Segmentations" # 包含所有的标签
    dst_image_dir = "val_nii_06/JPEGImages"
    dst_label_dir = "val_nii_06/Segmentations"
    
    file_names = os.listdir(source_image_dir)
    random.shuffle(file_names) # 打乱数据名
    for i in range(1000):
        file_name = file_names[i]
        source_jpeg_path = os.path.join(source_image_dir,file_name)
        dst_jpeg_path = os.path.join(dst_image_dir,file_name)
        shutil.move(source_jpeg_path,dst_jpeg_path)
        
        source_png_path = os.path.join(source_label_dir,file_name)
        dst_png_path = os.path.join(dst_label_dir,file_name)
        shutil.move(source_png_path,dst_png_path)