如何在Pygame的Surface对象中为指定颜色设置白名单

Pygame 没有内置的颜色白名单原生接口，你可以通过以下两种方案实现仅保留白名单颜色、其余颜色自动透明的需求：

方案1：逐像素遍历修改（适合小尺寸素材，比如图标） #

逻辑直观，直接遍历所有像素判断是否在白名单内，不符合则将alpha值设为0（完全透明）：

import pygame

def apply_color_whitelist(surface, whitelist):
    # 确保Surface带有alpha通道
    surface = surface.convert_alpha()
    # 锁定Surface提升像素操作性能
    surface.lock()
    
    for x in range(surface.get_width()):
        for y in range(surface.get_height()):
            # 取当前像素的RGB值，忽略原有alpha值
            rgb = surface.get_at((x, y))[:3]
            if rgb not in whitelist:
                # 不在白名单则设为透明
                surface.set_at((x, y), (*rgb, 0))
    
    surface.unlock()
    return surface

# 调用示例
icon = pygame.image.load("icon.png").convert()
# 白名单格式为 (R,G,B) 元组，用set存储提升查询效率
whitelist = {(255,255,255), (0,0,0), (255,0,0)}
processed_icon = apply_color_whitelist(icon, whitelist)

优缺点：实现简单易调试，但大尺寸素材遍历速度较慢，不适合每帧重复调用。

方案2：掩码批量处理（性能更优，适合大尺寸/高频调用场景） #

调用Pygame底层的掩码接口实现批量处理，性能远高于逐像素遍历：

import pygame

def apply_color_whitelist_mask(surface, whitelist):
    surface = surface.convert_alpha()
    w, h = surface.get_size()
    # 新建全透明的结果Surface
    res_surf = pygame.Surface((w, h), pygame.SRCALPHA)
    
    for color in whitelist:
        # 生成仅保留当前白名单颜色的临时Surface
        temp = surface.copy()
        # 把当前白名单颜色设为colorkey生成掩码
        temp.set_colorkey(color, pygame.RLEACCEL)
        mask = pygame.mask.from_surface(temp, 0)
        # 将当前颜色的有效区域绘制到结果Surface
        mask.to_surface(res_surf, source=surface, unsetcolor=(0,0,0,0))
    
    return res_surf

注意事项 #

• 操作前调用convert_alpha()确保Surface带有alpha通道，否则透明设置不会生效
• 白名单用set类型存储，成员查询的时间复杂度为O(1)，比list效率高很多
• 对静态素材建议提前缓存处理后的结果，避免重复计算消耗性能

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