将列表保存为python中的图像

from PIL import Image

def save_list_as_image(lst, width, height, output_path):
    # 创建一个新的图像对象
    image = Image.new("RGB", (width, height))

    # 将列表中的数据写入图像的像素
    for i in range(width):
        for j in range(height):
            # 获取列表中对应位置的像素值
            pixel = lst[i * height + j]

            # 将像素值写入图像的像素
            image.putpixel((i, j), pixel)

    # 保存图像
    image.save(output_path)

# 示例使用
lst = [(255, 0, 0), (0, 255, 0), (0, 0, 255)]  # 列表中的数据为RGB颜色值
width = 3  # 图像的宽度
height = 1  # 图像的高度
output_path = "image.png"  # 图像保存的路径

save_list_as_image(lst, width, height, output_path)