Python修改PNG图像像素值映射后保存未生效问题排查

当前需要处理仅包含1、255两类像素值的PNG格式文件，需实现的像素映射规则为：

• 像素值为255的像素修改为0
• 剩余值为1的像素修改为2

初始编写的实现代码如下：

for img_name in good_imgs:
  img = Image.open(img_name)
  pixels = img.load() 
  for i in range(img.size[0]):
    for j in range(img.size[1]):
        if pixels[i,j] == 255:
          pixels[i,j] = 0
  img.save(img_name)

for img_name in bad_imgs:
  img = Image.open(img_name)
  pixels = img.load() 
  for i in range(img.size[0]):
    for j in range(img.size[1]):
        if pixels[i,j] == 255:
          pixels[i,j] = 0
        elif pixels[i,j] == 1:
          pixels[i,j] == 2
  img.save(img_name)

运行代码后保存的图像像素值与原始文件完全一致，经单张测试确认像素修改逻辑本身可正常生效，初步定位问题出在img.save()保存环节。

• 语法错误：处理bad_imgs的分支中，值为1的像素修改逻辑写为pixels[i,j] == 2，这里误用了比较运算符==而非赋值运算符=，该分支的赋值操作从未实际执行。
• 逻辑缺失：处理good_imgs的分支中，仅实现了255转0的规则，遗漏了1转2的要求，不符合预设的映射规则。
• 图像模式兼容问题：保存后像素值无变化的核心原因是PNG文件常为索引模式（P模式）或1位二值模式（1模式），这类图像的像素值与内置调色板绑定，直接修改像素值后覆盖保存原文件时，PIL会自动基于原调色板做值优化；若写入的新值不在原调色板范围内，保存时会被自动修正回原值，导致修改失效。单张测试时如果手动转换过图像模式、或另存为新路径文件，就不会触发这个自动还原逻辑。

首先统一处理逻辑，打开图像后先转为8位灰度（L）模式，脱离原调色板限制，保存时关闭PNG优化选项，避免自动值压缩：

from PIL import Image

# 合并两组图片路径，避免重复写逻辑
all_imgs = good_imgs + bad_imgs
for img_path in all_imgs:
    # 打开后直接转L模式，消除索引/二值模式的调色板限制
    img = Image.open(img_path).convert("L")
    pixels = img.load()
    img_w, img_h = img.size
    for i in range(img_w):
        for j in range(img_h):
            val = pixels[i, j]
            if val == 255:
                pixels[i, j] = 0
            elif val == 1:
                pixels[i, j] = 2
    # 保存时关闭优化，防止自动调整像素值
    img.save(img_path, optimize=False)

如果图片分辨率较高、批量处理量大，可以用numpy做向量化运算替换逐像素循环，处理速度更快：

import numpy as np
from PIL import Image

all_imgs = good_imgs + bad_imgs
for img_path in all_imgs:
    img = Image.open(img_path).convert("L")
    img_arr = np.array(img)
    # 批量执行像素映射
    img_arr[img_arr == 255] = 0
    img_arr[img_arr == 1] = 2
    new_img = Image.fromarray(img_arr, mode="L")
    new_img.save(img_path, optimize=False)

校验方式：保存完成后重新打开对应图像，输出像素值的唯一集合，确认结果仅包含0、2两个值即可。

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