Python 基础(十五):枚举

1 简介

起初 Python 中并未内置枚举(enum)类型,枚举是在 Python3.4 添加的新功能,此时我们可能会有一个疑问:Python3.4 之前的版本还能不能使用枚举呢?

Python 基础(十五):枚举

答案是可以使用,但是不能直接使用,使用之前需要先用 pip install enum 安装。

什么是枚举?

枚举可看作是一系列符号名称的集合,集合中每一个元素要保证唯一性和不可变,因此我们可以对枚举中元素进行恒等比较,通俗来讲枚举就是一系列常量的集合,枚举是可迭代的。

枚举有什么作用?

我们先来思考一个问题:不使用枚举我们如何定义常量呢?

Python 基础(十五):枚举

常用的做法是采用变量名大写的方式来定义,这种方式虽然简单,但问题在于我们定义的仍然是变量、是可以被修改的,而常量是什么呢?简单来说就是不可变的量,枚举就有不可变的特性,所以枚举的主要作用就是用来定义常量的。

2 使用

2.1 创建

枚举语法与 class 语法相同,之前我们在Python 基础(十一):面向对象中已经介绍过 class 了,枚举的定义可以通过继承 Enum 的方式来实现, 看一下示例:

  •  
from enum import Enum
class WeekDay(Enum): Mon = 0 Tue = 1 Wed = 2 Thu = 3 Fri = 4

2.2 访问

枚举成员及属性的访问如下所示:

  •  
# 枚举成员print(WeekDay.Mon)# 枚举成员名称print(WeekDay.Mon.name)# 枚举成员值print(WeekDay.Mon.value)

枚举的迭代也很简单,如下所示:

  •  
# 方式 1for day in WeekDay:    # 枚举成员    print(day)    # 枚举成员名称    print(day.name)    # 枚举成员值    print(day.value)
# 方式 2print(list(WeekDay))

2.3 比较

枚举成员及属性可以使用 is 进行对象比较,还可以使用 == 进行值比较,看下示例:

  •  
print(WeekDay.Mon is WeekDay.Thu)print(WeekDay.Mon == WeekDay.Mon)print(WeekDay.Mon.name == WeekDay.Mon.name)print(WeekDay.Mon.value == WeekDay.Mon.value)

枚举成员不能进行大小比较,如下所示:

  •  
>>> WeekDay.Mon < WeekDay.ThuTypeError: '<' not supported between instances of 'WeekDay' and 'WeekDay'

2.4 确保枚举值唯一

我们定义枚举时,成员名称是不可以重复的,但成员值是可以重复的,如果想要保证成员值不可重复,可以通过装饰器 @unique 来实现,如下所示:

  •  
 from enum import Enum, unique
@uniqueclass WeekDay(Enum): Mon = 0 ...

 

Python 基础(十五):枚举

 

上一篇:如何为LSTM重新构建输入数据(Keras)


下一篇:读懂POLARDB不能错过的18篇深度文章!