合并两个文件夹下名称交集的标签

标签为黑白图，合并两个文件夹下名称交集的标签

 

# 合并两个文件夹下相同名称的两张png标签
#
# 开发时间：2023/5/18 16:38
import os

from PIL import Image

def merge(path1, path2, path3):
    # 打开第一张图片
    img1 = Image.open(path1)
    # 打开第二张图片
    img2 = Image.open(path2)
    # 获取图片的大小
    width, height = img1.size
    # 创建新的图片对象
    new_img = Image.new('L', (width, height), 0)
    # 合并两张图片
    for x in range(width):
        for y in range(height):
            pixel1 = img1.getpixel((x, y))
            pixel2 = img2.getpixel((x, y))
            if pixel1 == 255 or pixel2 == 255:
                new_img.putpixel((x, y), 255)
            else:
                new_img.putpixel((x, y), 0)

    file_name = os.path.split(path1)[1]

    path_new = os.path.join(path3, file_name)
    # 保存新的图片
    new_img.save(path_new)


# 提取需要合并的同名文件
A_file_path = 'Ajson'
A_fileList = os.listdir(A_file_path)

B_file_path = 'Bjson'
B_fileList = os.listdir(B_file_path)

Alist = []
for file_a in A_fileList:
    Alist.append(file_a)

Blist = []
for file_b in B_fileList:
    Blist.append(file_b)

# 将列表转化为集合，并计算交集
common_set = set(Alist).intersection(set(Blist))

# 将交集转换为列表
common_list = list(common_set)

if len(common_list) == 0:
    print("这两个列表没有相同的元素")
else:
    print("这两个列表有相同的元素：")
    print(len(common_list))

    for item in common_list:
        print(item)
    print(len(common_list))


# 拼接目录，遍历合并
for item in common_list:

    path_a = os.path.join(A_file_path, item)
    path_b = os.path.join(B_file_path, item)
    path_result = 'result'
    if not os.path.exists(path_result):
        os.makedirs(path_result)

    merge(path_a, path_b, path_result)