自訂類型的枚舉

但有些時候我們需要控制枚舉的型別,那麼我們可以 Enum 衍生出自定義類別來滿足這種需求。透過修改上面的例子:

#!/usr/bin/env python3
# -*- coding: UTF-8 -*-
from enum import Enum, unique
Enum('Month', ('Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun', 'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec'))
# @unique 装饰器可以帮助我们检查保证没有重复值
@unique
class Month(Enum):
    Jan = 'January'
    Feb = 'February'
    Mar = 'March'
    Apr = 'April'
    May = 'May'
    Jun = 'June'
    Jul = 'July'
    Aug = 'August'
    Sep = 'September '
    Oct = 'October'
    Nov = 'November'
    Dec = 'December'
if __name__ == '__main__':
    print(Month.Jan, '----------',
          Month.Jan.name, '----------', Month.Jan.value)
    for name, member in Month.__members__.items():
        print(name, '----------', member, '----------', member.value)

輸出的結果如下:

7ee49a517fbabe7772b6f5c5b285ad0.png

#透過上面的例子,可以知道枚舉模組定義了具有迭代(interator)和比較(comparison) 功能的枚舉類型。它可以用來為值創建明確定義的符號,而不是使用具體的整數或字串。

繼續學習