使用Python在数据中寻找需要图片并保存

已有一部分修改过的图片，需要在原数据中找到对应的原文件并保存。

import os
from PIL import Image

data_contrast_dir = "D:\\对比\\1106B,yphd" #存放对比图片地址
data_base_dir = "D:\\对比\\数据"           #存放原始图片数据

contrast = [] #建立列表，用于保存图片信息
base = []

contrast_number = 0     #对比图片数目
base_number = 0         #原始图片文件夹数目
number = 0              #检索过文件夹数目


for file_contrast in os.listdir(data_contrast_dir):     #构件对比图片列表
    contrast.append(file_contrast[0:32])                #file_contrast为data_contrast_dir当前目录下图片名
    contrast_number = contrast_number + 1

for file_base in os.listdir(data_base_dir):             #构件原始图片文件夹信息列表
    base.append(file_base) 
    base_number = base_number + 1

for a in range(base_number):                            #循环原始图片文件夹数目次
    a = base.pop()                                      #读取原始图片数据列表
    c = a[0:32]                                             
    number = number+1       
    if c in contrast:                                    #进行对比
        for file_d in os.listdir(data_base_dir + "\\" + a):
            if file_d.endswith("NG.bmp"):
                img = Image.open(data_base_dir + "\\" + a + "\\" + file_d)
                img.save("D:\\对比\\1\\" + file_d)        #图片保存目录
                
    print("进度：" + str(number) + "/" + str(base_number))