Python枚举类型教程_enum模块使用详解

Python enum模块提供类型安全、可迭代、不可变的命名常量枚举类,支持名称/值双向查找、auto()自动赋值、@unique防别名、属性方法扩展及IntEnum整数比较。

Python 的 enum 模块提供了一种清晰、安全、可读性强的方式来定义一组命名常量。它不是简单地用变量代替数字,而是构建出有类型、可迭代、不可变、支持名称/值双向查找的枚举类。

定义与基本访问方式

enum 导入 Enum,继承后定义类成员即可:

  • 每个成员由名称(.name)和值(.value)组成,二者严格分离
  • 通过点号访问:Color.RED 返回枚举成员对象
  • 获取名称字符串:Color.RED.name"RED"
  • 获取关联值:Color.RED.value1
  • 按名称字符串查找:Color["RED"] 等价于 Color.RED
  • 按值查找:Color(1) 返回 Color.RED(自动映射,不创建新实例)

三种常用遍历方法

枚举类本身可直接迭代,返回的是规范成员(排除别名),顺序按定义顺序:

  • 直接 for 循环:最直观,适合逐个处理成员
    for color in Color: print(color.name, color.value)
  • 转为列表:获取所有成员对象组成的列表
    list(Color)[, , ...]
  • 显式提取名称或值集合:如 [c.name for c in Color][c.value for c in Color]

自动赋值与别名处理

避免手动编号重复,可用 auto() 自动生成递增整数:

  • from enum import Enum, auto
  • class Status(Enum): PENDING = auto(); PAID = auto() → 值分别为 1、2
  • 相同值的成员会被视为别名(如 BLUE = 1YELLOW = 1),BLUE is YELLOWTrue
  • 若需禁止别名,用 @unique 装饰器,重复值会触发 ValueError

扩展功能:带属性与方法的枚举

枚举成员可以携带额外数据,并封装逻辑:

  • __init__ 中接收构造参数,绑定实例属性
  • 使用 @property 定义只读计算属性(如行星质量、半径推导重力加速度)
  • 支持自定义 __str____repr__ 提升可读性
  • 若需支持整数比较或隐式转换,改用 IntEnum 继承

不复杂但容易忽略:枚举类不能被实例化,成员不可修改,也不支持大小比较(> 会报 TypeError),仅支持身份(is)和相等(==)判断。