python实现心型照片墙效果

from PIL import Image
import os
import math

map = [
    [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1],
    [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1],
    [1, 1, 1, 1, 1, 0, 0, 0, 1, 1, 1, 1, 1, 0, 0, 0, 1, 1, 1, 1, 1],
    [1, 1, 1, 1, 0, 0, 0, 0, 0, 1, 1, 1, 0, 0, 0, 0, 0, 1, 1, 1, 1],
    [1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1],
    [1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1],
    [1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1],
    [1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1],
    [1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1],
    [1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1],
    [1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1],
    [1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1],
    [1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1],
    [1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1],
    [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1],
    [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1],
    [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1],
]

# 这个是我图片的所在路径
img_dir = r"C:\Users\admin\Desktop\头像"
# 获取这个文件下的所有文件，因为没有过滤其他类型文件，所以不要放非图片文件
# 返回的是一个列表
imgs = os.listdir(img_dir)
# 设置图片的尺寸，所有图片尺寸要保持统一，要有组织有纪律
img_h = img_w = 192
# 计算行数，即子列表的个数
rows = len(map)
# 计算列数，即子列表中元素的个数
columns = len(map[0])

# 第一个参数照着写
# 第二个参数需要传入一个元组，元组的第一个参数是画布的宽，第二个是高
# 第三个参数传入的是画布的颜色
# 使用Image.new()方法创建一个画布
figure = Image.new("RGB", (img_w*columns, img_h*rows),"white")

# 表示图片的下标
count = 0
# 遍历行
for i in range(len(map)):
    # 遍历每行中的所有元素
    for j in range(len(map[i])):
        # 如果元素是1，就不管它
        if map[i][j] == 1:
            continue
        # 如果元素是非1，即0就放图片上去
        else:
            # 做个异常处理，防止有些图片打开失败，导致程序中断
            try:
                # 使用Image.open("图片路径")方法获取图片对象
                image = Image.open(os.path.join(img_dir, imgs[count]))
            except:
                continue
            # resize((新的宽，新的高))用来改变图片的尺寸,接收一个元组
            image = image.resize((img_w, img_h))
            # 将修改尺寸后的图片(image)粘贴(paste)到画布(figure)上
            # 第一个参数 是图片对象
            # 第二个参数是 图片在画布上的位置，相当于单元格的位置
            figure.paste(image, (img_w*j, img_h*i))
            # 使用完一张图片就要记录下来，并开始使用下一张图片
            count += 1
        
# 当循环结束，即表示心型照片图已经完成
# 将画好的画布显示出来，给你看到
figure.show()
# 不满意在调整二维数组中0的位置，满意可以保存到本地，用来装逼
# 需要告知程序图片保存的路径
figure.save('心型照片墙.png')