Python3 enum – 枚举类型
在Python 3.4及以上版本,可以使用内置的enum
模块来创建枚举类型。枚举是一种特殊的类,其中每个元素都是唯一的成员。
下面是一个使用enum
模块创建枚举的例子:
from enum import Enum
class Color(Enum):
RED = 1
GREEN = 2
BLUE = 3
# 使用枚举
def print_color(color):
if color == Color.RED:
print("Red")
elif color == Color.GREEN:
print("Green")
elif color == Color.BLUE:
print("Blue")
print_color(Color.RED) # 输出: Red
print_color(Color.GREEN) # 输出: Green
print_color(Color.BLUE) # 输出: Blue
在这个例子中,我们定义了一个名为Color
的枚举类,其中包含三个成员:RED
、GREEN
和BLUE
。每个成员都有一个唯一的值,这些值可以是整数或者其他不同的类型。在print_color
函数中,我们通过比较枚举成员来决定打印什么颜色。这种方式使得代码更加清晰和易于维护。
评论已关闭