Python 的枚舉 Enum

枚舉是常用的功能,看看Python的枚舉.

from enum import Enum

Month = Enum('Month', ('Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun', 'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec'))

 

@unique
class Weekday(Enum):
    Sun = 0 # Sun的value被設定爲0
    Mon = 1
    Tue = 2
    Wed = 3
    Thu = 4
    Fri = 5
    Sat = 6

 

for name, member in Month.__members__.items():
    print(name, '=>', member, ',', member.value)

 

枚舉的定義

1. 首先，定義枚舉要導入enum模塊。
2. 枚舉定義用class關鍵字，繼承Enum類。

注意:

　　定義枚舉時，成員名稱不允許重複　

　　默認情況下，不同的成員值允許相同。但是兩個相同值的成員，第二個成員的名稱被視作第一個成員的別名　

　　　如果枚舉中存在相同值的成員，在通過值獲取枚舉成員時，只能獲取到第一個成員

　如果要限制定義枚舉時，不能定義相同值的成員。可以使用裝飾器@unique【要導入unique模塊】


for name, member in Month.__members__.items():
    print(name, '=>', member, ',', member.value)

我們就獲得了Month類型的枚舉類，可以直接使用Month.Jan來引用一個常量，或者枚舉它的所有成員.

訪問這些枚舉類型可以有若干種方法:

枚舉支持迭代器，可以遍歷枚舉成員

 

>>> day1 = Weekday.Mon
>>> print(day1)
Weekday.Mon
>>> print(Weekday.Tue)
Weekday.Tue
>>> print(Weekday['Tue'])
Weekday.Tue
>>> print(Weekday.Tue.value)
2
>>> print(day1 == Weekday.Mon)
True
>>> print(day1 == Weekday.Tue)
False
>>> print(Weekday(1))
Weekday.Mon
>>> print(day1 == Weekday(1))
True
>>> Weekday(7)
Traceback (most recent call last):
  ...
ValueError: 7 is not a valid Weekday
>>> for name, member in Weekday.__members__.items():
...     print(name, '=>', member)
...
Sun => Weekday.Sun
Mon => Weekday.Mon
Tue => Weekday.Tue
Wed => Weekday.Wed
Thu => Weekday.Thu
Fri => Weekday.Fri
Sat => Weekday.Sat

枚舉取值 總結:

通過成員的名稱來獲取成員;通過成員值來獲取成員;通過成員，來獲取它的名稱和值.

 

注意: Enum 的成員均爲單例（Singleton），並且不可實例化，不可更改.

枚舉是可以比較的:

舉成員可進行同一性比較,可進等值比較,不能進行大小比較.

總結:Enum可以把一組相關常量定義在一個class中，且class不可變，而且成員可以直接比較,並且枚舉有多種實現方法。