python3 enum模塊的應(yīng)用實例詳解

更新時間：2019年08月12日 14:06:56 作者：達標(biāo)

這篇文章主要介紹了python3 enum模塊的應(yīng)用，文中提到了字典類型的缺點及特點，本文通過實例代碼給大家介紹的非常詳細，具有一定的參考借鑒價值,需要的朋友可以參考下

一、枚舉與字典類型

字典類型的缺點：

1.值可變
2.沒有防止相同標(biāo)簽的功能

枚舉的特點：
1.枚舉類的值不可以被外界更改
2.不能存在相同的標(biāo)簽，但允許不同標(biāo)簽存在相同的枚舉值，即后者相當(dāng)于前者的別名
3.枚舉值可以是任意類型
4.枚舉標(biāo)簽盡量用大寫

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 #枚舉值不能動態(tài)更改
>>>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 #不能出現(xiàn)相同的標(biāo)簽
>>>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.獲取枚舉類型中的標(biāo)簽：.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)  #枚舉的標(biāo)簽
print(VIP.YELLOW.value)  #枚舉的數(shù)值
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 #枚舉標(biāo)簽與枚舉值比較，并不相等
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

四、枚舉的轉(zhuǎn)換

class VIP(Enum):
  RED = 1
  GREEN = 'green'
  YELLOW = True
  BLUE = 2
print(VIP(2))  #通過具體值獲取枚舉類型，返回VIP.BLUE
print(VIP('green'))  #返回VIP.GREEN

五、枚舉的拓展unique

unique:規(guī)定兩個不同的枚舉不能取相同的值

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自動生成對應(yīng)的數(shù)值

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值是調(diào)用_generate_next_value_()函數(shù)來產(chǎn)生的，我們可以通過重寫該函數(shù)

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))  #轉(zhuǎn)換列表
>>>
[<VIP.RED: 'RED'>, <VIP.GREEN: 'GREEN'>, <VIP.YELLOW: 'YELLOW'>, <VIP.BLUE: 'BLUE'>]

總結(jié)

以上所述是小編給大家介紹的python3 enum模塊的應(yīng)用實例詳解,希望對大家有所幫助，如果大家有任何疑問請給我留言，小編會及時回復(fù)大家的。在此也非常感謝大家對腳本之家網(wǎng)站的支持！
如果你覺得本文對你有幫助，歡迎轉(zhuǎn)載，煩請注明出處，謝謝！

您可能感興趣的文章: