python—批量从文件夹随机抽取图片

代码实现：

从n个文件夹中，分别随机抽取m图片，剪切到新的n个文件夹。

import os,sys
import random
import shutil


def copyFile(fileDir):
    pathDir = os.listdir(fileDir)
    sample = random.sample(pathDir, 300)
    print(sample)
    for name in sample:
        shutil.move(fileDir + name, tarDir + name)


if __name__ == '__main__':
    # open /textiles
    path = "/home/fairy/workspace/dataset/textiles/"
    dirs = os.listdir(path)
    i = 0
    # output all folds
    for file in dirs:
        print(file)
        i = i+1
        filename = "/home/fairy/workspace/dataset/Fabric" + str(i)
        os.mkdir(filename)
        fileDir = path + "Fabric"+str(i) + "/"
        tarDir = filename + "/"
        copyFile(fileDir)

 

今日感慨：

Python太好用了吧！！！

我爱Python Python爱我