Python中枚举类型的详解（代码示例）(2)-木庄网络博客

当前第2页返回上一页

from enum import Enum, unique
@unique
class Color(Enum):
    red   = 1
    green = 2
    blue  = 1  # ValueError: duplicate values found in <enum 'Color'>: blue -> red

枚举取值

可以通过成员名来获取成员也可以通过成员值来获取成员:

print(Color['red'])  # Color.red  通过成员名来获取成员
 
print(Color(1))      # Color.red  通过成员值来获取成员

每个成员都有名称属性和值属性：

member = Color.red
print(member.name)   # red
print(member.value)  # 1

支持迭代的方式遍历成员，按定义的顺序，如果有值重复的成员，只获取重复的第一个成员：

1 2	`for` `color in Color:` `print(color)`

特殊属性 __members__ 是一个将名称映射到成员的有序字典，也可以通过它来完成遍历：

1 2	`for` `color in Color.__members__.items():` `print(color) # ('red', <Color.red: 1>)`

枚举比较

枚举的成员可以通过 is 同一性比较或通过 == 等值比较：

Color.red is Color.red
Color.red is not Color.blue
 
Color.blue == Color.red
Color.blue != Color.red

枚举成员不能进行大小比较：

1	`Color.red < Color.blue # TypeError: unorderable types: Color() < Color()`

扩展枚举 IntEnum

IntEnum 是 Enum 的扩展，不同类型的整数枚举也可以相互比较：

from enum import IntEnum
class Shape(IntEnum):
    circle = 1
    square = 2
 
class Request(IntEnum):
    post = 1
    get = 2
 
print(Shape.circle == 1)            # True
print(Shape.circle < 3)             # True
print(Shape.circle == Request.post) # True
print(Shape.circle >= Request.post) # True