Python 枚举【一】

1. 枚举的定义

  1. 首先,定义枚举要导入 enum 模块。
  2. 枚举定义用 class 关键字,继承 Enum 类。
  3. 用于定义枚举的 class 和定义类的 class 是有区别【下一篇博文继续分享】。

  示例代码:

from enum import Enum

class
Color(Enum): red = 1 orange = 2 yellow = 3 green = 4 blue = 5 indigo = 6 purple = 7

  代码分析:

  1. 上面的代码,我们定义了颜色的枚举 Color.
  2. 颜色枚举有 7 个成员,分别是 Color.red、Color.orange、Color.yellow 等。
  3. 每一个成员都有它们各自名称和值,Color.red 成员的名称是:red,值是:1。
  4. 每个成员的数据类型就是它所属的枚举。【* 注:用 class 定义的类,实际上就是一种类型】
1.1 定义枚举时,成员名称不允许重复   
from enum import Enum

class
Color(Enum): red = 1 red = 2

  上面的代码,就无法执行。提示错误:TypeError: Attempted to reuse key: 'red'

 1.2 默认情况下,不同的成员值允许相同。但是两个相同值的成员,第二个成员的名称被视作第一个成员的别名  

from enum import Enum

class
Color(Enum): red = 1 red_alias = 1

  成员 Color.red 和 Color.red_alias 具有相同的值,那么成员 Color.red_alias 的名称 red_alias 就被视作成员 Color.red 名称 red 的别名。

   1.3 如果枚举中存在相同值的成员,在通过值获取枚举成员时,只能获取到第一个成员

from enum import Enum

class
Color(Enum): red = 1 red_alias = 1

print(Color(1))

  输出结果为:Color.red

 1.4 如果要限制定义枚举时,不能定义相同值的成员。可以使用装饰器 @unique【要导入 unique 模块】

from enum import Enum, unique

@unique
class Color(Enum):
red
= 1
red_alias
= 1

  再执行就会提示错误:ValueError: duplicate values found in <enum 'Color'>: red_alias -> red

 

2. 枚举取值 

 2.1 通过成员的名称来获取成员

Color['red']

 2.2 通过成员值来获取成员

Color(2)

 2.3 通过成员,来获取它的名称和值

red_member = Color.red
red_member.name
red_member.value

 

3. 迭代器

 3.1 枚举支持迭代器,可以遍历枚举成员

for color in Color:
    print(color)

  输出结果是,枚举的所有成员。Color.red、Color.orange、Color.yellow、Color.green、Color.blue、Color.indigo、Color.purple。

 3.2 如果枚举有值重复的成员,循环遍历枚举时只获取值重复成员的第一个成员

from enum import Enum

class Color(Enum):
red
= 1
orange
= 2
yellow
= 3
green
= 4
blue
= 5
indigo
= 6
purple
= 7
red_alias
= 1

for color in Color:
print(color)

  输出结果是:Color.red、Color.orange、Color.yellow、Color.green、Color.blue、Color.indigo、Color.purple。但是Color.red_alias并没有出现在输出结果中。

 3.3 如果想把值重复的成员也遍历出来,要用枚举的一个特殊属性 __members__

from enum import Enum

class Color(Enum):
red
= 1
orange
= 2
yellow
= 3
green
= 4
blue
= 5
indigo
= 6
purple
= 7
red_alias
= 1

for color in Color.members.items():
print(color)

  输出结果:('red', <Color.red: 1>)、('orange', <Color.orange: 2>)、('yellow', <Color.yellow: 3>)、('green', <Color.green: 4>)、('blue', <Color.blue: 5>)、

       ('indigo', <Color.indigo: 6>)、('purple', <Color.purple: 7>)、('red_alias', <Color.red: 1>)

 

4. 枚举比较

 4.1 枚举成员可进行同一性比较

Color.red is Color.red

  输出结果是:True

Color.red is not Color.blue

  输出结果是:True

 4.2 枚举成员可进等值比较

Color.blue == Color.red

  输出结果是:False

Color.blue != Color.red

  输出结果是:True

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

Color.red < Color.blue

  输出结果出错:TypeError: unorderable types: Color()< Color()