在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。
本文链接:https://my.lmcjl.com/post/18131.html
4 评论