摘要：起步的原生类型中并不包含枚举类型。枚举类型可以看作是一种标签或是一系列常量的集合，通常用于表示某些特定的有限集合，例如星期月份状态等。简单的示例枚举成员有值默认可重复，枚举成员具有友好的字符串表示枚举类型不可实例化，不可更改。
起步
Python 的原生类型中并不包含枚举类型。为了提供更好的解决方案，Python 通过 PEP 435 在 3.4 版本中添加了 enum 标准库。
枚举类型可以看作是一种标签或是一系列常量的集合，通常用于表示某些特定的有限集合，例如星期、月份、状态等。在没有专门提供枚举类型的时候我们是怎么做呢，一般就通过字典或类来实现：
Color = {
"RED" : 1,
"GREEN": 2,
"BLUE" : 3,
}
class Color:
RED = 1
GREEN = 2
BLUE = 3
这种来实现枚举如果小心翼翼地使用当然没什么问题，毕竟是一种妥协的解决方案。它的隐患在于可以被修改。
使用 Enum
更好的方式是使用标准库提供的 Enum 类型，官方库值得信赖。3.4 之前的版本也可以通过 pip install enum 下载支持的库。简单的示例：
from enum import Enum
class Color(Enum):
red = 1
green = 2
blue = 3
枚举成员有值（默认可重复），枚举成员具有友好的字符串表示：
>>> print(Color.red)
Color.red
>>> print(repr(Color.red))
>>> type(Color.red)
>>> isinstance(Color.green, Color)
True
枚举类型不可实例化，不可更改。
定义枚举
定义枚举时，成员名不允许重复
class Color(Enum):
red = 1
green = 2
red = 3 # TypeError: Attempted to reuse key: "red"
成员值允许相同，第二个成员的名称被视作第一个成员的别名
class Color(Enum):
red = 1
green = 2
blue = 1
print(Color.red) # Color.red
print(Color.blue) # Color.red
print(Color.red is Color.blue)# True
print(Color(1)) # Color.red 在通过值获取枚举成员时，只能获取到第一个成员
若要不能定义相同的成员值，可以通过 unique 装饰
from enum import Enum, unique
@unique
class Color(Enum):
red = 1
green = 2
blue = 1 # ValueError: duplicate values found in : blue -> red
枚举取值
可以通过成员名来获取成员也可以通过成员值来获取成员:
print(Color["red"]) # Color.red 通过成员名来获取成员
print(Color(1)) # Color.red 通过成员值来获取成员
每个成员都有名称属性和值属性：
member = Color.red
print(member.name) # red
print(member.value) # 1
支持迭代的方式遍历成员，按定义的顺序，如果有值重复的成员，只获取重复的第一个成员：
for color in Color:
print(color)
特殊属性 __members__ 是一个将名称映射到成员的有序字典，也可以通过它来完成遍历：
for color in Color.__members__.items():
print(color) # ("red", )
枚举比较
枚举的成员可以通过 is 同一性比较或通过 == 等值比较：
Color.red is Color.red
Color.red is not Color.blue
Color.blue == Color.red
Color.blue != Color.red
枚举成员不能进行大小比较：
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
总结
enum 模块功能很明确，用法也简单，其实现的方式也值得学习，有机会的话可以看看它的源码。
文章版权归作者所有，未经允许请勿转载,若此文章存在违规行为，您可以联系管理员删除。
转载请注明本文地址：https://www.ucloud.cn/yun/42782.html
