使用python将图像的多个部分连接到单个图像

from PIL import Image

def join_images(images, direction):
    """
    将多个图像按照指定方向连接在一起
    :param images: 图像列表
    :param direction: 连接方向，'horizontal'表示横向拼接，'vertical'表示纵向拼接
    :return: 拼接后的图像
    """
    if direction == 'horizontal':
        width = sum(img.width for img in images)
        height = max(img.height for img in images)
    elif direction == 'vertical':
        width = max(img.width for img in images)
        height = sum(img.height for img in images)
    else:
        raise ValueError("Invalid direction. Must be 'horizontal' or 'vertical'.")

    result = Image.new('RGB', (width, height))

    offset_x = 0
    offset_y = 0
    for img in images:
        result.paste(img, (offset_x, offset_y))
        if direction == 'horizontal':
            offset_x += img.width
        elif direction == 'vertical':
            offset_y += img.height

    return result

# 示例用法
image1 = Image.open('image1.jpg')
image2 = Image.open('image2.jpg')
image3 = Image.open('image3.jpg')

joined_image_horizontal = join_images([image1, image2, image3], 'horizontal')
joined_image_horizontal.save('joined_image_horizontal.jpg')

joined_image_vertical = join_images([image1, image2, image3], 'vertical')
joined_image_vertical.save('joined_image_vertical.jpg')