使用pillow制作长图

这是来自一个妹子的需求，需要将多张图片拼接成一张长图

我是使用pillow这个库来实现的，下面的简单的代码，操作比较简单，代码还有优化的空间

def test(dirpath):
    ims = [Image.open(os.path.join(dirpath, i)) for i in os.listdir(dirpath)]

    height_s = 0
    for i in range(len(ims)):
        height_s += ims[i].size[1]

    res = Image.new(ims[0].mode, (ims[0].size[0], height_s))

    height = 0
    for i,j in enumerate(ims): 
        res.paste(j, box=(0, height))
        height += j.size[1]

    res.save('res.jpg')