python移动文件：将一个文件夹中的特定后缀名的图片移动到另一个文件夹

写在前面

在进行数据集的调用时，我发现别人的数据集的目录格式跟我不太一样。我不太想改变dataset中的代码。所以我就把图片移到我想要的目录结构中

开干

1、os.walk(top[, topdown=True[, οnerrοr=None[, followlinks=False]]]) ‘’’

os.walk(top[, topdown=True[, onerror=None[, followlinks=False]]])
top - - 是你所要遍历的目录的地址, 返回的是一个三元组(root, dirs, files)。
root所指的是当前正在遍历的这个文件夹的本身的地址
dirs 是一个list ，内容是该文件夹中所有的目录的名字(不包括子目录)
files同样是 list, 内容是该文件夹中所有的文件(不包括子目录)

‘’’

import os

from PIL import Image

nowpath='D:/CsdnProject/datasets/train'

newpath='D:/CsdnProject/datasets/Mask_bi'
#将训练里的mask移到Mask_bi中
def transferPictures(nowpath, newpath):
    # 将文件夹下的不同类别的文件夹中的部分图片转移到另一个文件夹下的相同类别的文件夹下，并删除原文件夹中的相应图片（类似于剪切）

#shift+tab整体缩进
a='abcdefg'
if(a[-4:-1]=='def'):
    print(a[:-1])
def transferPictures(nowpath,newpath):
    for roots, dirs, files in os.walk(nowpath):
        #print("111")
        for imgname in files:
            #print('222')
            if imgname[-5:-1]=='k.pn':
                imgpath = nowpath  + '/' + imgname
                # print(imgpath)
                for roots, dirs, files in os.walk(newpath):
                    im = Image.open(imgpath)
                    im.save(newpath  + '/' + imgname)
                    os.remove(imgpath)  # 转移完后删除原图片


transferPictures(nowpath, newpath)

我的目录结构

其中imgname[-5:-1]==‘k.pn’:这句代码就是找到你想要的相关图片，这里我写的比较简陋。意思就是后缀为这个的全都移动到新的目录中。