欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

python3 enum模块的应用

程序员文章站 2024-01-13 17:46:10
python枚举模块的学习 ps:小编刚开始学习没多久,部分资源来源于其他网友,如有出错,麻烦联系修改哈,互帮互助,共同进步 一、枚举与字典类型 字典类型的缺点:1.值可变 2.没有防止相同标签的功能 枚举的特点:1.枚举类的值不可以被外界更改 2.不能存在相同的标签,但允许不同标签存在相同的枚举值 ......

python枚举模块的学习

ps:小编刚开始学习没多久,部分资源来源于其他网友,如有出错,麻烦联系修改哈,互帮互助,共同进步

一、枚举与字典类型

字典类型的缺点:1.值可变 2.没有防止相同标签的功能

枚举的特点:1.枚举类的值不可以被外界更改 2.不能存在相同的标签,但允许不同标签存在相同的枚举值,即后者相当于前者的别名 3.枚举值可以是任意类型 4.枚举标签尽量用大写

from enum import enum  #普通类

class dict():
    green = 1
    green = 2
    red = 3

dict.red = 4   
print(dict.red)

>>> 4

 

 

class vip(enum): 
    red = 1
    green =2
    yellow = 3

vip.green = 5  #枚举值不能动态更改

>>>file "d:\python\lib\enum.py", line 361, in __setattr__

  raise attributeerror('cannot reassign members.')

  attributeerror: cannot reassign members.
class vip(enum):
    red = 1
    green =2
    yellow = 3
    yellow = 3  #不能出现相同的标签

>>>file "d:\python\lib\enum.py", line 92, in __setitem__
       raise typeerror('attempted to reuse key: %r' % key)
       typeerror: attempted to reuse key: 'yellow'
class vip(enum):
    red = 1
    green =2
    yellow = 3
    blue = 3

print(vip.blue)

>>>vip.yellow   #即vip.blue类似于vip.yellow的别名
class vip(enum):
    red = 1  #整型
    blue = 1.2 #浮点型
    green = 'green'  #str
    yellow = true  #布尔值

二、枚举类型、枚举名称、枚举值

1.获取枚举类型中的标签:.name属性

2.获取枚举类型中的值:.value属性

from enum import enum
class vip(enum):
    red = 1
    green = 'green'
    yellow = true
    blue = 1.2

print(vip.red)  #枚举的类型
print(vip.yellow.name)    #枚举的标签
print(vip.yellow.value)    #枚举的数值
print('遍历成员。。。')
for i in vip.__members__.items():    #枚举的遍历
    print(i)

>>>
vip.red
red
1
遍历成员。。。
('red', <vip.red: 1>)
('green', <vip.green: 'green'>)
('yellow', <vip.red: 1>)
('blue', <vip.blue: 1.2>)

三、枚举的比较运算

class vip(enum):
    red = 1
    green = 'green'
    yellow = true
    blue = 2
#枚举的比较
a = vip.red == vip.red  #枚举值可以做等值比较
b = vip.red == vip.green
c = vip.red is vip.red
d = vip.red == 1  #枚举标签与枚举值比较,并不相等
f = vip.red < vip.blue  #枚举值不支持大小比较,打印会报错typeerror: '<' not supported between instances of 'vip' and 'vip'

print(a)
print(b)
print(c)
print(d)

>>>
true
false
true
false

四、枚举的转换

class vip(enum):
    red = 1
    green = 'green'
    yellow = true
    blue = 2

print(vip(2))   #通过具体值获取枚举类型,返回vip.blue
print(vip('green'))    #返回vip.green

五、枚举的拓展unique

unique:规定两个不同的枚举不能取相同的值

from enum import enum,unique

@unique
class vip(enum):
    red = 1
    green = 1
    yellow = true
    blue = 2

>>>报错
valueerror: duplicate values found in <enum 'vip'>: green -> red, yellow -> red

六、枚举的拓展auto

如果我们对枚举值没有特别的要求或者没有太大的影响时,我们可以使用auto自动生成对应的数值

 

from enum import enum,auto

class vip(enum):
    red = auto()
    green = auto()
    yellow = auto()
    blue = auto()

for i in vip.__members__.items():   #遍历枚举
    print(i)

>>>
('red', <vip.red: 1>)
('green', <vip.green: 2>)
('yellow', <vip.yellow: 3>)
('blue', <vip.blue: 4>)

 

其实,auto值是调用_generate_next_value_()函数来产生的,我们可以通过重写该函数

from enum import enum,auto

class auto_test(enum):
    def _generate_next_value_(name, start, count, last_values):
        return name

class vip(auto_test):
    red = auto()
    green = auto()
    yellow = auto()
    blue = auto()

# for i in vip.__members__.items():  #遍历枚举
#     print(i)
print(list(vip))   #转换列表

>>>
[<vip.red: 'red'>, <vip.green: 'green'>, <vip.yellow: 'yellow'>, <vip.blue: 'blue'>]