按比例随机抽取文件夹下的图片，将抽取的图片移动到新的文件夹下

import os
import random
import shutil

def moveFile(fileDir, saveDir):
    pathDir = os.listdir(fileDir)  # 图片原始路径
    filenumber = len(pathDir)
    rate1 = 0.2  # 自定义抽取csv文件的比例
    picknumber1 = int(filenumber * rate1)  # 按照rate比例从文件夹中取一定数量的文件
    sample1 = random.sample(pathDir, picknumber1)  # 随机选取
    print(sample1)
    for name in sample1:
        shutil.move(fileDir + name, saveDir + "\\" + name)
    return

if __name__ == '__main__':
    fileDir = "E:\管制物品检测数据集\images\\"  #抽取图片文件夹下的路径
    saveDir = 'E:\\test1' #保存被抽取图片的路径
    moveFile(fileDir, saveDir)