2022年 11月 7日

python中枚举类的理解

文章目录

  • 前言
  • 一、枚举是什么?
  • 二、为何要使用枚举类(好处)?
    • 1、为了代码可读性
    • 2、为了不被修改
    • 3、为了不被继承
  • 三、如何定义及使用枚举类?
    • 1.Enum关键字:
    • 2.定义一个更精确的枚举类
    • 3.如何访问枚举类中的常量(即成员)
  • 四、枚举类和普通类的区别?
  • 五、总结

前言

欢迎指正讨论:

读了两篇文章后总结了关于python中枚举类的理解,可以先看此篇,觉得不够再去看以下的参考文:

1、廖雪峰的python之枚举类
2、枚举到底是个啥


`提示:以下是本篇文章正文内容

一、枚举是什么?

广义的理解:可以是一个可被一一列举的集合。简单举个列子:

周一到周日这就是一个枚举;一月到十二月也是一个枚举。

枚举的两个特性:

  • 可被列举的集合(不能是无穷无尽的)
  • 不可变性 (一周7天我有生之年应该不会变了)

二、为何要使用枚举类(好处)?

1、为了代码可读性

#枚举类
from enum import Enum, unique
@unique    #此装饰器可以帮助我们检查保证没有重复值
class WeekDay1(Enum):
    Sun = 0
    Mon = 1
    Tue = 2
    Wed = 3
    Thu = 4
    Fri = 5
    Sat = 6
print(WeekDay1.Sat.value)   #代码一看就知道打印的是周六对应的值
print(6)                   #你能看出来代码背后的意义吗
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13

上述代码中,一看 print(WeekDay.Sat.value) 就知道打印的是周六对应的值,而 print(6) 你能一眼看出来么?

2、为了不被修改

还是用上面代码例子为例:

#枚举类
from enum import Enum, unique
@unique    #此装饰器可以帮助我们检查保证没有重复值
class WeekDay1(Enum):
    Sun = 0
    Mon = 1
    Tue = 2
    Wed = 3
    Thu = 4
    Fri = 5
    Sat = 6
print(WeekDay1.Sat.value)   #代码一看就知道打印的是周六对应的值
#WeekDay1.Sun = 7   #由于重新被赋值,报错

#普通类
class WeekDay2(object):
    Sun = 0
    Mon = 1
    Tue = 2
    Wed = 3
    Thu = 4
    Fri = 5
    Sat = 6
print(WeekDay2.Sat)
WeekDay2.Sun = 7    #普通类直接修改

#报错,无法继承枚举类WeekDay1
#class WeekDay3(WeekDay1):
#   pass

  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
  • 28
  • 29
  • 30

上述代码中想修改属性对应的值,会直接报错;而普通类的修改则ok。

3、为了不被继承

这一点很好理解,因为为了不让修改,自然而然也就不让继承了,代码上面已经给了。


三、如何定义及使用枚举类?

1.Enum关键字:

Python提供了Enum类来定义:

from enum import Enum

Month = Enum('Month', ('Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun', 'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec'))   
#第一个参数是类名,第二个参数是需要枚举的常量集合
  • 1
  • 2
  • 3
  • 4

这样我们就获得了Month类型的枚举类。可以直接使用Month.Jan来引用一个常量,或者枚举它的所有成员(具体的访问枚举常量见第三点):

for name, member in Month.__members__.items():
    print(name, '=>', member, ',', member.value)
""" Output:
Jan => Month.Jan , 1
Feb => Month.Feb , 2
Mar => Month.Mar , 3
Apr => Month.Apr , 4
May => Month.May , 5
Jun => Month.Jun , 6
Jul => Month.Jul , 7
Aug => Month.Aug , 8
Sep => Month.Sep , 9
Oct => Month.Oct , 10
Nov => Month.Nov , 11
Dec => Month.Dec , 12 """
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15

value属性则是自动赋给成员的int常量,默认从1开始计数

2.定义一个更精确的枚举类

如果需要更精确地控制枚举类型,可以从Enum派生出自定义类,此时value值不再默认从1开始:

#更精确定义的一个枚举类,即value值不从默认的1开始
#枚举类
from enum import Enum, unique
@unique    #此装饰器可以帮助我们检查保证没有重复值
class WeekDay1(Enum):
    Sun = 0
    Mon = 1
    Tue = 2
    Wed = 3
    Thu = 4
    Fri = 5
    Sat = 6
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12

3.如何访问枚举类中的常量(即成员)

首先清楚一点,枚举类里都有哪些家庭成员:成员名(Sun)、成员(WeekDay1.Sun)、成员对应的值(0)

#更精确定义的一个枚举类,即value值不从默认的1开始
#枚举类
from enum import Enum, unique
@unique    #此装饰器可以帮助我们检查保证没有重复值
class WeekDay1(Enum):
    Sun = 0
    Mon = 1
    Tue = 2
    Wed = 3
    Thu = 4
    Fri = 5
    Sat = 6
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 遍历所有常量

    #遍历 成员名、成员、成员对应的值
    for name, member in WeekDay1.__members__.items():
        print(name, '=>', member, ',', member.value)
    """ Output:
    Sun => WeekDay1.Sun , 0
    Mon => WeekDay1.Mon , 1
    Tue => WeekDay1.Tue , 2
    Wed => WeekDay1.Wed , 3
    Thu => WeekDay1.Thu , 4
    Fri => WeekDay1.Fri , 5
    Sat => WeekDay1.Sat , 6 """
    
    • 1
    • 2
    • 3
    • 4
    • 5
    • 6
    • 7
    • 8
    • 9
    • 10
    • 11
  • 访问某个具体常量

    • 通过成员名访问
    #成员名访问枚举常量
    tmp = WeekDay1.Mon
    print(tmp)
    #Output: WeekDay1.Mon
    
    print(WeekDay1['Mon'])
    #Output: WeekDay1.Mon
    
    • 1
    • 2
    • 3
    • 4
    • 5
    • 6
    • 7
    • 通过value值访问
    #value值访问枚举常量
    print(WeekDay1(1)) 
    #Output: WeekDay1.Mon
    
    • 1
    • 2
    • 3

四、枚举类和普通类的区别?

1、枚举类中的value无法被修改,而普通类可以
2、从Enum派生的枚举类无非被继承,而普通类则可以被继承
3、访问时不一样,这点一定注意

#枚举类
from enum import Enum, unique
@unique    #此装饰器可以帮助我们检查保证没有重复值
class WeekDay1(Enum):
    Sun = 0
    Mon = 1
    Tue = 2
    Wed = 3
    Thu = 4
    Fri = 5
    Sat = 6
#普通类
class WeekDay2(object):
    Sun = 0
    Mon = 1
    Tue = 2
    Wed = 3
    Thu = 4
    Fri = 5
    Sat = 6
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20

1、枚举类中的value无法被修改,而普通类可以

#WeekDay1.Sun = 7   #枚举类WeekDay1中的value由于重新被赋值,报错
WeekDay2.Sun = 7    #普通类WeekDay1中的value可以被重新赋值
  • 1
  • 2

2、从Enum派生的枚举类无非被继承,而普通类则可以被继承

#报错,无非继承枚举类WeekDay1
#class WeekDay3(WeekDay1):
#   pass

#可以继承普通类WeekDay2,不报错
class WeekDay3(WeekDay2):
   pass
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7

3、访问时不一样,这点一定注意

print(WeekDay1.Sat)    #Output: WeekDay1.Sat,注意这里输出不是‘6’!!!
print(WeekDay2.Sat)    #Output: 6
print(WeekDay1.Sat.value)   #Output: 6
  • 1
  • 2
  • 3

五、总结

三连,感谢!