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
函数来复制文件