【Python】将文件夹内不同特定名称的文件分开存档

需求

• 文件夹内有大量图片，图片格式均为jpg，文件名最后的部分有若干种，要将每种分开存入不同的文件夹。

import os
import shutil

pin_image_folder = r'D:\test_pics\2bsent\pin' #新的文件路径
solder_image_folder = r'D:\test_pics\2bsent\solder'
aoi_image_folder = r'D:\test_pics\2bsent\aoi'

dir_path = r'D:\test_pics\raw_img'  #存放所有图像的路径

for root, dirs, files in os.walk(dir_path):#遍历读取文件夹筛选符合标准的图片
     # root 表示当前正在访问的文件夹路径     # dirs 表示该文件夹下的子目录名list     # files 表示该文件夹下的文件list
    for file in files:
         if file.count('PIN_PASS.jpg') == 1:
            shutil.copy(os.path.join(root,file), pin_image_folder)     #复制图片到新文件夹
            # os.remove(os.path.join(root,file))    #删除名称带有'.jpg'和'.JPG'的图像
          elif file.count('Soldering_PASS.jpg') == 1:
             shutil.copy(os.path.join(root, file), solder_image_folder)
          elif file.count('AOI_PASS.jpg') == 1:
             shutil.copy(os.path.join(root, file), aoi_image_folder)