Python从多个文件夹中随机复制图片

从多个文件夹中随机复制图片，例如src有10个文件夹，每个子文件夹有500张图，下文代码为从src中每个文件夹随机选取60张图并复制到dir目录下。

 

import os
import random
import shutil


def moveFile(fileSrc, fileDir, pickNumber):
    subSrcNames = os.listdir(fileSrc)
    subFileNumbers = len(subSrcNames)   #源路径下子文件夹数量

    for index in range(0, subFileNumbers):
        if (os.path.exists(fileDir + subSrcNames[index]) == False):  # 如果目标目录没有该子集，创建文件夹
            os.mkdir(fileDir + subSrcNames[index])
        imgaeSubDir = os.listdir(fileDir + subSrcNames[index])
        for images in imgaeSubDir:  # 如过目标目录存在子文件夹，删除原有目标目录子文件夹下的图片
            os.remove(fileDir + subSrcNames[index] + "\\" + images)

        imgaeSubSrc = os.listdir(fileSrc + subSrcNames[index] + '/') #如果源目录图片数量不够，则选择的图片数量=源目录图片数量
        imageNumSubSrc = len(imgaeSubSrc)
        pickNum = pickNumber
        if (imageNumSubSrc < pickNum):
            pickNum = imageNumSubSrc
        selSampleIndex = random.sample(imgaeSubSrc, pickNum)  # 随机选取源目录中pickNum个样本

        for imageName in selSampleIndex:
            shutil.copy(fileSrc + subSrcNames[index] + "\\" + imageName,
                        fileDir + subSrcNames[index] + "\\" + imageName)

    return


if __name__ == '__main__':
    filePthSrc = r"C:/Users/asus/Desktop/src/"
    filePthDir = r"C:/Users/asus/Desktop/dir/"
    pickNumber = 60
    moveFile(filePthSrc, filePthDir, pickNumber)