将一个文件夹中的多个子文件夹中的图像存储在新的文件夹中

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) # 递归地将文件或目录移动到另一个位置

 

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