python将两张图片横向或者纵向合成一张

import os

import cv2
import numpy as np
import pygame
from PIL import Image


def compose_two_image_with_hstack(left, right):
    img1 = cv2.imread(left)
    img2 = cv2.imread(right)
    image = np.hstack((img1, img2))
    cv2.imwrite("left_right.png", image)


def compose_two_image_with_vstack(up, down):
    img1 = cv2.imread(up)
    img2 = cv2.imread(down)
    image = np.vstack((img1, img2))
    cv2.imwrite("up_down.png", image)


if __name__ == '__main__':
    compose_two_image_with_hstack("left.png", "right.png")
    compose_two_image_with_vstack("up.png", "down.png")