You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

遍历图像对象列表保存文件仅输出最后一张图像的问题如何解决

问题产生原因

  • 保存文件时重复使用了同一个文件名变量:你代码里的f是最开始遍历os.listdir('.')时的循环变量,遍历结束后f只会保留最后一个遍历到的文件名,后续保存所有图片时都用这个同名变量,每一次保存都会覆盖上一次的文件,最终自然只会剩下最后一张图片。
  • 初始逻辑遗漏了文件名与图片的对应关系:你只把打开的Image对象存入了列表,没有同时保留原文件名,也没有为打乱后的图片生成独立的新文件名,无法为每张图片生成唯一的存储路径。
  • 初始遍历逻辑存在风险:直接遍历当前目录所有文件,会把非图片文件、甚至后续新建的目标文件夹也当成图片读取,很容易触发异常。

修复后的代码

import os
from PIL import Image
import random

def makemydir(whatever):
    try:
        os.makedirs(whatever, exist_ok=True)
    except OSError:
        pass
    # 不直接切换工作目录,后续用完整路径存储更稳妥,避免路径逻辑混乱

# 筛选常见图片格式,避免读入非图片文件报错
valid_suffix = {'jpg', 'jpeg', 'png', 'bmp', 'webp'}
img_list = []
for f in os.listdir('.'):
    # 跳过目录,只处理文件
    if os.path.isfile(f):
        suffix = f.split('.')[-1].lower()
        if suffix in valid_suffix:
            # 同时存储原文件名和图片对象,方便后续灵活处理命名规则
            img_list.append((f, Image.open(f)))

# 随机打乱图片顺序
random.shuffle(img_list)

target_dir = input('What is the target directory name \n')
makemydir(target_dir)

# 按新顺序生成唯一文件名存储,避免覆盖
for idx, (origin_name, img) in enumerate(img_list):
    print(f"正在保存:{origin_name} -> {idx+1}.jpg")
    # 如果需要保留原文件名,可替换为下面的写法
    # img.save(os.path.join(target_dir, origin_name))
    img.save(os.path.join(target_dir, f"{idx+1}.jpg"))

print("全部图片保存完成")

核心修改点说明

  • 新增了文件类型校验逻辑,只读取符合格式的图片文件,跳过目录和非图片文件,避免运行报错。
  • 存储图片时同步记录原文件名,可自主选择按新顺序生成有序文件名,或保留原文件名存储。
  • 取消了切换工作目录的操作,改用路径拼接的方式生成存储路径,避免多场景下路径逻辑混乱。
  • 为每个打乱后的图片生成独立的唯一文件名,彻底解决文件覆盖问题。

内容的提问来源于stack exchange,提问作者Elroy 1624

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.09.27 08:54:04