【python】将文件夹下图片随机抽取一定比例,并复制到新文件夹

import os
import random
import shutil

def moveFile(fileDir,tarDir):
    pathDir = os.listdir(fileDir)  # 取图片的原始路径
    filenumber = len(pathDir)
    rate = 0.1  # 自定义抽取图片的比例,比方说100张抽10张,那就是0.1
    picknumber = int(filenumber * rate)  # 按照rate比例从文件夹中取一定数量图片
    sample = random.sample(pathDir, picknumber)  # 随机在pathDir选取picknumber数量的样本图片
    print(sample)
    for name in sample:
        shutil.copyfile(fileDir + name, tarDir + name)
    return

org_img_folder = 'E:\\夏甸迁移\\未知单元成矿指标_流体指标\\控矿指标_流体指标tiff数据new\\控矿指标_流体指标tiff数据new\\'
tar_img_folder = 'E:\\夏甸迁移\\data\\unknown\\unknow\\'
moveFile(org_img_folder,tar_img_folder)

shutil.move 函数移动文件,源文件会被删除或移动到目标位置。

shutil.copyfile 函数来复制文件

你可能感兴趣的:(python)