Python枚举类

前言

枚举类是一些具有特殊含义的类,比如用一个类表示月份,则该类的实例对象最多有 12 个,一周则有7个对象,Python 3.4 中新增加了 Enum 枚举类,对于这些实例化对象个数固定的类,可以用枚举类来定义。

枚举类定义

  • 只需要令其继承自 enum 模块中的 Enum 类,则证明这是一个枚举类
  • 枚举类的每个成员都由 2 部分组成,分别为 name 和 value
  • name不允许重复,value允许重复
  • 和普通类的用法不同,枚举类不能用来实例化对象
  • 枚举类中各个成员的值,不能在类的外部做任何修改

创建方式一

from enum import Enum # 第一个参数用于指定枚举类的类名,第二个参数用于指定枚举类中的多个成员。 Colors = Enum(Colors, ('red', 'green', 'blue')) # 未指定值,默认从 1 开始递增 print(Colors.red.value) print(Colors.green.value) print(Colors.blue.value) ### 运行结果 1 2 3 

创建方式二

from enum import Enum class Color(Enum):     # 为序列值指定value值     red = 1     green = 2     blue = 3 	 #调用枚举成员的 3 种方式 print(Color.red) print(Color['red']) print(Color(1)) #调取枚举成员中的 value 和 name print(Color.red.value) print(Color.red.name) #遍历枚举类中所有成员的 2 种方式 for color in Color:     print(color)  ### 运行结果: Color.red Color.red Color.red 1 red Color.red Color.green Color.blue 

使用

1.正常使用

print(Color.red == Color.green) print(Color.red.name is Color.green.name) ### 运行结果 Flase Flase 

2.不能在类的外部做任何修改,下面语法的做法是错误的:

Color.red = 4 报错:     raise AttributeError('Cannot reassign members.') AttributeError: Cannot reassign members. 

3.枚举类中各个成员必须保证 name 互不相同,但 value 可以相同,举个例子:

class Color(Enum):     # 为序列值指定value值     red = 1     green = 1     blue = 3 print(Color['green'])  ### 运行结果 Color.red 

可以看到,Color 枚举类中 red 和 green 具有相同的值(都是 1),Python 允许这种情况的发生,它会将 green 当做是 red 的别名,因此当访问 green 成员时,最终输出的是 red。

4.设置value唯一性,不允许出现相同的value

#引入 unique from enum import Enum,unique #添加 unique 装饰器 @unique class Color(Enum):     # 为序列值指定value值     red = 1     green = 1     blue = 3 print(Color['green'])  ### 运行结果报错: Traceback (most recent call last):   File test.py, line 3, in <module>     class Color(Enum):   File /python3.6/lib/enum.py, line 834, in unique     (enumeration, alias_details)) ValueError: duplicate values found in <enum 'Color'>: green -> red