Python 定義枚舉

首先需要安裝模塊 enum34

pip install enum34

1 定義枚舉類型

from enum import Enum
class Color(Enum):
red =1
green =2
blue =3

取枚舉值的方式：

print(Color.red)

輸入：Color.red ，直接使用枚舉屬性值取值

print(Color(1))

輸出：Color.red， 使用索引取值

print(Color[‘green’])

輸出：Color.green ， 使用枚舉名稱索引

2. 遍歷枚舉值

x = [c for c in Color]
print(x)

輸出： [<Color.red: 1>, <Color.green: 2>, <Color.blue: 3>]