將一個文件夾中的多個子文件夾中的圖像存儲在新的文件夾中

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

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章