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) # 遞歸地將文件或目錄移動到另一個位置