import os, random, shutil
def moveFile(fileDir,tarDir):
data_list = []
pathDir = os.listdir(fileDir)
for i in range(len(pathDir)):
if pathDir[i].split(".")[1] == 'png':
data_list.append(pathDir[i])
sample = random.sample(data_list, 1)
for name in sample:
shutil.move(fileDir + name, tarDir + name)
return
if __name__ == '__main__':
for i in range(1,51):
fileDir = "/Path_Datasource/images_"+str("%03d"%(i))+"/left_images/"
tarDir = "/Path_Datasaved/images_"+str("%03d"%(i))+"/left_colorimages/" # 移动到新的文件夹路径
moveFile(fileDir,tarDir)
print(i,"ok!")
代码学习:
import os
random.sample(list, N) # 从list中随机获取N个元素
import shutil
shutil.move(fileDir + name, tarDir + name) # 递归地将文件或目录移动到另一个位置