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

96 阅读 0 评论 64 点赞

我是靠谱客的博主安静丝袜，这篇文章主要介绍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)

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
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)