Python枚举类型使用详解_enum模块应用示例【指导】

Python 的 enum 模块提供安全、可读、类型友好的具名常量定义方式;相比普通变量,Enum 防止重复赋值、拼写错误和越界取值,支持迭代、比较、序列化及 IDE 补全。

Python 的 enum 模块提供了一种定义具名常量集合的规范方式,比用普通变量或字典更安全、可读性更强,也支持迭代、比较和序列化等操作。

为什么要用 Enum 而不是普通变量?

直接用字符串或数字定义状态(如 STATUS_ACTIVE = 1)容易出错:可能重复赋值、拼写错误、无法限制取值范围。Enum 强制所有合法值必须显式声明,且每个成员是唯一的对象实例,天然支持类型检查和 IDE 自动补全。

  • 成员不可变,不能被重新赋值
  • 不同枚举类的同名成员不会冲突(Color.REDState.RED 是独立的)
  • 支持 is 判断、== 比较、in 成员检测

基础用法:定义和访问枚举成员

继承 Enum 类即可创建枚举类。每个属性是一个枚举成员,自动获得唯一序号(从 1 开始),也可显式赋值。

from enum import Enum

class Color(Enum): RED = 1 GREEN = 2 BLUE = 3

访问方式

print(Color.RED) # Color.RED print(Color.RED.name) # 'RED' print(Color.RED.value) # 1 print(Color.RED.value == 1) # True print(Color(2)) # Color.GREEN(按 value 查找) print(Color['BLUE']) # Color.BLUE(按 name 查找)

常用枚举变体:IntEnum、StrEnum、auto()

IntEnum 允许与整数直接比较;StrEnum(Python 3.11+)让成员值默认为字符串且支持字符串比较;auto() 自动递增赋值,避免手动编号出错。

from enum import IntEnum, StrEnum, auto

class Priority(IntEnum): LOW = 10 MEDIUM = 20 HIGH = 30

print(Priority.HIGH > Priority.LOW) # True(IntEnum 支持数值比较)

class Status(StrEnum): PENDING = auto() # 自动设为 'PENDING' DONE = auto() # 自动设为 'DONE'

print(Status.PENDING == "PENDING") # True

实用技巧:遍历、验证和 JSON 序列化

枚举类本身可迭代,适合用于下拉选项或状态校验;转 JSON 需自定义编码器,通常推荐存 namevalue 字段。

  • 遍历所有成员:list(Color)for c in Color:
  • 检查是否为有效成员:value in Color.__members__.values() 或用 try/except 捕获 ValueError
  • JSON 序列化示例:
    import json
    class Encoder(json.JSONEncoder):
        def default(self, obj):
            if isinstance(obj, Enum):
                return obj.name  # 或 obj.value
            return super().default(obj)
    

    json.dumps({'color': Color.RED}, cls=Encoder) # '{"color": "RED"}'