如何编写img.save()按列表索引命名保存numpy数组转存的图片

首先你需要用Pillow库把numpy数组转换成PIL图像对象，然后通过循环遍历列表，结合索引生成文件名即可。具体步骤如下：

1. 导入依赖库 #

import numpy as np
from PIL import Image

2. 遍历列表并按索引保存 #

用enumerate()函数可以同时拿到每个元素的索引和对应的numpy数组，然后按索引生成带编号的文件名：

# 假设你的numpy数组列表是image_list
for idx, img_array in enumerate(image_list):
    # 注意：如果numpy数组是float类型（取值0-1），需要先转换为uint8类型（取值0-255）
    if img_array.dtype == np.float32 or img_array.dtype == np.float64:
        img_array = (img_array * 255).astype(np.uint8)
    
    # 将numpy数组转为PIL图像
    img = Image.fromarray(img_array)
    
    # 按索引命名保存，比如保存为image_0.jpeg、image_1.jpeg...
    save_path = f"image_{idx}.jpeg"
    img.save(save_path, format='jpeg')

注意事项 #

• 如果你的numpy数组是RGB格式，直接用Image.fromarray()即可；如果是灰度图，可能需要指定模式，比如Image.fromarray(img_array, mode='L')
• 确保numpy数组的取值范围符合图像要求：uint8类型是0-255，float类型是0-1（转换时要乘255转成uint8）

内容的提问来源于stack exchange，提问作者lu xin