1. 程式人生 > >Python 的列舉 Enum

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不可變,而且成員可以直接比較,並且列舉有多種實現方法。