Python中如何以地道写法检查值是否存在于Enum枚举值集合中

原生Enum类默认的__contains__方法仅支持校验枚举成员是否属于当前枚举类（例如Color.RED in Color会返回True），不支持直接校验传入的原始值，你当前使用的any(s == c.value for c in Color)写法虽然能实现需求，但每次判断都要遍历全量枚举项，时间复杂度为O(n)，枚举项较多时性能一般，也不够简洁，更推荐以下几种方案：

• 方案1：使用Enum内置映射表做常数时间查找 #

  所有Enum类在定义阶段会自动维护_value2member_map_类属性，这是一个存储「枚举值 -> 枚举成员」映射关系的字典，直接判断目标值是否存在于该映射中即可，查找效率为O(1)：

  from enum import Enum
  
  class Color(str, Enum):
      RED = "red"
      GREEN = "green"
      YELLOW = "yellow"
  
  s = "red"
  if s in Color._value2member_map_:
      print(Color(s))
  

  虽然这个属性以下划线开头，但属于Enum模块从首个版本就保留的稳定实现，不属于随时可能变动的私有内部逻辑，可以放心使用。

• 方案2：采用EAFP风格的异常捕获 #

  这是Python社区最推崇的实现方式，遵循「先执行、出错再处理」的EAFP（Easier to ask for forgiveness than permission）原则，不需要提前做存在性校验，直接尝试用目标值构造枚举实例，捕获值不合法时抛出的ValueError即可，兼容所有Python3版本，也不依赖任何内部属性：

  s = "red"
  try:
      color = Color(s)
      print(color)
  except ValueError:
      # 处理值不存在的逻辑
      pass
  

• 方案3：封装通用基类适配高频校验场景 #

  如果你需要在多个枚举类中频繁做值存在性校验，可以自定义一个带通用校验方法的Enum基类，后续业务枚举直接继承即可复用逻辑：

  from enum import Enum
  
  class BaseValueEnum(Enum):
      @classmethod
      def has_value(cls, val):
          return val in cls._value2member_map_
  
  class Color(str, BaseValueEnum):
      RED = "red"
      GREEN = "green"
      YELLOW = "yellow"
  
  s = "red"
  if Color.has_value(s):
      print(Color(s))
  

注：Python 3.11+ 版本提供了内置的StrEnum类，替代手动继承str, Enum的写法，但值校验逻辑和上述方案完全通用，没有差异。

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