这里可以移动文件或者文件夹:
# 深度学习过程中,需要制作训练集和验证集、测试集。
import os, random, shutil
def moveFile(fileDir):
pathDir = os.listdir(fileDir) # 取文件的原始路径
filenumber=len(pathDir)
rate=0.1 # 自定义抽取文件的比例,比方说100个文件抽10个,那就是0.1
picknumber=int(filenumber*rate) # 按照rate比例从文件夹中取一定数量文件
sample = random.sample(pathDir, picknumber) # 随机选取picknumber数量的样本文件
print (sample)
for name in sample:
shutil.move(fileDir+name, tarDir+name)
return
if __name__ == '__main__':
fileDir = "./source/" # 源文件夹路径
tarDir = './result/' # 移动到新的文件夹路径
moveFile(fileDir)