Python中的枚举类型示例介绍_Python

Python中的枚举类型示例介绍

2021-05-13 00:27栖迟于一丘 Python

这篇文章主要给大家介绍了关于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