实战 | 用 Python 轻松制作好看的心型照片墙
俊红的数据分析之路
共 7125字，需浏览 15分钟
·
2021-09-13 10:54
一、效果
四、代码
1、导库
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],
]
3、导入图片
# 这个是我图片的所在路径
img_dir = r"C:\Users\admin\Desktop\头像"
# 获取这个文件下的所有文件，因为没有过滤其他类型文件，所以不要放非图片文件
# 返回的是一个列表
imgs = os.listdir(img_dir)
# 设置图片的尺寸，所有图片尺寸要保持统一，要有组织有纪律
img_h = img_w = 192
# 计算行数，即子列表的个数
rows = len(map)
# 计算列数，即子列表中元素的个数
columns = len(map[0])
4、画图
# 第一个参数照着写
# 第二个参数需要传入一个元组，元组的第一个参数是画布的宽，第二个是高
# 第三个参数传入的是画布的颜色
# 使用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')
评论