Python PIL将PNG非透明像素改为黑色的代码错误排查

你当前的实现有三个核心错误，导致处理结果不符合预期：

• 透明判断逻辑完全错误：带透明通道的PNG像素通过getpixel()获取到的是(R, G, B, A)四元组，其中最后一位A（alpha通道）才是控制透明度的参数，A=0代表完全透明，和RGB三个通道的色值没有任何关系。你写的判断条件data[0]!=255 and data[1]!=0 and data[2]!=0完全没有读取alpha值，靠RGB色值筛选像素，必然会出现大量漏改、错改的情况，比如纯红色像素会因为满足data[1]==0的条件，不会被转换为黑色。
• 像素写入时丢失透明信息：你调用putpixel()时传入的是(0,0,0)三元组，没有保留原像素的alpha值，哪怕判断逻辑修正，也会破坏原有的透明效果，把半透明、全不透明像素的透明度信息覆盖掉。
• 逐像素调用getpixel/putpixel的写法效率极低，处理大尺寸图片时运行速度会非常慢。

逐像素易读版本 #

逻辑和你原有写法一致，修正判断和写入逻辑即可：

from PIL import Image

img_da_maskerare = Image.open("temp.png").convert("RGBA")
width, height = img_da_maskerare.size
pixels = img_da_maskerare.load()

for i in range(width):
    for j in range(height):
        r, g, b, a = pixels[i, j]
        # 只要alpha值大于0（非完全透明），就改成黑色，保留原有透明度
        if a > 0:
            pixels[i, j] = (0, 0, 0, a)

img_da_maskerare.show()
# 保存结果用：img_da_maskerare.save("output.png")

高性能批量处理版本 #

利用PIL的通道操作接口，不用手动遍历像素，运行速度更快：

from PIL import Image

img = Image.open("temp.png").convert("RGBA")
# 拆分出原图像的alpha通道
*_, alpha = img.split()
# 创建和原图尺寸一致的全黑图层，和原alpha通道合并得到结果
black_rgb = Image.new("RGB", img.size, (0, 0, 0))
result = Image.merge("RGBA", (*black_rgb.split(), alpha))

result.show()

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