python 照片墙

import os
from PIL import Image
import matplotlib.pyplot as plt

files=os.listdir("images1")

images=[]
for file in files:
    img=Image.open('images1/'+file)
    images.append(img)

#设置心形点阵
style=[[0,1,0,1,0],
       [1,1,1,1,1],
       [1,1,1,1,1],
       [0,1,1,1,0],
       [0,0,1,0,0]]       

#创建画板
col=5
row=5
board=Image.new('RGB',(200*col,200*row),'gray')

for i in range(row):
  for j in range(col):
    #判断点阵中如果为1的元素进行添加图片
    if style[i][j]==1:
      index=j+col*i
      #%号取余解决图片不足的问题    
      board.paste(images[index%len(images)],(20+j*200,20+200*i))

plt.imshow(board)
plt.show()
board.save("heart.png")