Python中的枚举类型示例介绍

起步

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))
<color.red: 1>
>>> type(color.red)
<enum 'color'>
>>> 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 <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

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

for color in color:
 print(color)

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

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

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

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 模块功能很明确，用法也简单，其实现的方式也值得学习，有机会的话可以看看它的源码。

好了，以上就是这篇文章的全部内容了，希望本文的内容对大家的学习或者工作具有一定的参考学习价值，如果有疑问大家可以留言交流，谢谢大家对的支持。