1. 程式人生 > >Python與列舉

Python與列舉

前言

列舉是一個常見的資料型別。在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