Python枚举的定义与3种使用方法

在Python中，枚举是一种数据类型，它是一组预定义的常量值，常常被用于代表有限集合的元素。Python3.4之后，官方引入了Enum类，使得枚举的定义和使用变得更加简单和直观。

Python 枚举的定义

在Python中定义枚举类非常简单，只需要继承Enum类即可。例如，定义一个表示星期几的枚举类可以这样写：

from enum import Enum

class Weekday(Enum):
    MON = 1
    TUE = 2
    WED = 3
    THU = 4
    FRI = 5
    SAT = 6
    SUN = 7

这里定义了一个名为Weekday的枚举类，其中包含了7个枚举值，分别对应星期一到星期日。枚举值的定义格式为name = value，其中name为枚举值的名称，value为枚举值的取值。

除了直接赋值之外，枚举值还可以通过其他方式来定义。例如，可以使用其他枚举值的取值来定义新的枚举值：

from enum import Enum

class Color(Enum):
    RED = 1
    GREEN = 2
    BLUE = 3
    WHITE = RED | GREEN | BLUE
    BLACK = 0

这里定义了一个名为Color的枚举类，其中WHITE的取值是RED | GREEN | BLUE，也就是将RED、GREEN和BLUE三个枚举值的取值进行按位或运算得到的结果。

Python 枚举的使用

Python 获取枚举值

在Python中，枚举值可以像普通变量一样使用。例如，可以通过枚举值的名称或者取值来获取枚举值：

>>> weekday = Weekday.MON
>>> print(weekday)
Weekday.MON
>>> print(weekday.name)
MON
>>> print(weekday.value)
1
>>> print(Weekday(2))
Weekday.TUE

可以使用name属性获取枚举值的名称，使用value属性获取枚举值的取值。另外，可以使用Enum(value)函数根据取值来获取对应的枚举值。

Python 枚举迭代

枚举类还可以进行迭代，获取枚举值的名称和取值：

>>> for day in Weekday:
...     print(day.name, day.value)
...
MON 1
TUE 2
WED 3
THU 4
FRI 5
SAT 6
SUN 7

Python 枚举比较与排序

枚举值还可以进行比较和排序，这是因为Python中的枚举值默认实现了eq和lt等比较运算符。
下面是一个简单的 Python 枚举值进行比较和排序的实例：

from enum import Enum

# 定义一个枚举类
class Color(Enum):
    RED = 1
    GREEN = 2
    BLUE = 3

# 比较两个枚举值
if Color.RED == Color.RED:
    print("Color.RED == Color.RED")
else:
    print("Color.RED != Color.RED")

if Color.RED == Color.GREEN:
    print("Color.RED == Color.GREEN")
else:
    print("Color.RED != Color.GREEN")

# 对枚举值进行排序
colors = [Color.GREEN, Color.BLUE, Color.RED]
sorted_colors = sorted(colors)

print("排序后的颜色值：")
for color in sorted_colors:
    print(color)

输出结果：

Color.RED == Color.RED
Color.RED != Color.GREEN
排序后的颜色值：
Color.RED
Color.GREEN
Color.BLUE

在这个例子中，我们首先定义了一个枚举类 Color，其中包含三个枚举值：RED、GREEN 和 BLUE。

然后我们比较了两个枚举值 Color.RED 和 Color.GREEN，由于它们不相等，因此输出了 Color.RED != Color.GREEN。

最后，我们创建了一个包含三个枚举值的列表 colors，并对它们进行排序。排序后，我们使用 for 循环遍历排序后的列表，并输出每个枚举值的名称。

需要注意的是，在 Python 3.4 之前，枚举类并没有 Enum 基类，而是继承自 object。如果你在旧版本的 Python 中使用枚举类，请确保将 Enum 替换为 object。