Python與列舉
阿新 • • 發佈:2018-12-24
前言
列舉是一個常見的資料型別。在Python中列舉卻不是一個型別,而是一個類。簡單粗暴!
使用
from enum import Enum
class VIP(Enum):
YELLOW = 2
GREEN = 2
BLACK = 4
RED = 5
print(type(VIP))
print(type(VIP.GREEN))
結果
<class 'enum.EnumMeta'>
<enum 'VIP'>
很簡單,很粗暴。但是列舉看起來跟字典有點像,為啥不用字典呢。因為這裡的列舉一旦做好了之後就不能動態的去修改。而字典是可以動態的去修改的。列舉一半作為資料對映有自己獨特的特點。
而且列舉的標籤是唯一的!不能出現相同的標籤。
標籤唯一但是數值不唯一。如果數值不唯一,相當於起別名。
我們的普通遍歷是直接去掉別名的。
例如:
from enum import Enum
class VIP(Enum):
GREEN = 2
GREEN_ALIAS = 2
BLACK = 4
RED = 5
for v in VIP:
print(v)
VIP.GREEN
VIP.BLACK
VIP.RED
如果你想獲取到、必須這樣遍歷到。
for v in VIP.__members__:
print(v)
如果你想數值也唯一!要使用unique註解。這裡如果不一樣就會報錯。
from enum import Enum,unique
@unique
class VIP(Enum):
GREEN = 1
GREEN_ALIAS = 2
BLACK = 4
RED = 5