Python 枚举唯一值

在Python中，枚举被用于表示一组固定的值。

定义枚举类

使用 enum 模块可以创建枚举类。

import enum

class Color(enum.Enum):
    RED = 1
    GREEN = 2
    BLUE = 3

此外，枚举类也可以直接继承 Enum：

from enum import Enum

class Color(Enum):
    RED = 1
    GREEN = 2
    BLUE = 3

访问枚举成员

可以使用点号操作符访问枚举成员：

print(Color.RED)

输出：

Color.RED

访问成员的名称和值：

print(Color.RED.name, Color.RED.value)

输出：

RED 1

遍历枚举类

可以使用 for 循环遍历枚举类：

for c in Color:
    print(c)

输出：

Color.RED
Color.GREEN
Color.BLUE

比较枚举成员

枚举成员可以使用 is 运算符进行比较：

print(Color.RED is Color.RED)
print(Color.RED is Color.GREEN)

输出：

True
False

应用枚举类

可以使用枚举类代替常量：

def print_color(color):
    if color == Color.RED:
        print('red')
    elif color == Color.GREEN:
        print('green')
    elif color == Color.BLUE:
        print('blue')

或者作为字典的键：

d = {
    Color.RED: 'red',
    Color.GREEN: 'green',
    Color.BLUE: 'blue',
}
print(d[Color.RED])

输出：

red