枚举是一种数据类型，在编程中用于表示一组相关的常量。枚举中的每个常量都有一个名称和一个对应的值，可以用于增强代码的可读性和可维护性。在Python中，枚举是由enum模块提供的，而不是Python提供专用的枚举语法。关于enum模块介绍见：enum。如需详细了解Python的enum模块，参见文章：Python enum和枚举(enum)介绍。
文章目录
- 1 语法介绍
- 2 参考
1 语法介绍
基础示例
# 引入 Enum 模块，用于创建枚举from enum import Enum# 创建一个枚举类Color，从Python内置的枚举类Enum继承class Color(Enum):# 定义 RED 数值为 1RED = 1# 定义 GREEN 数值为 2GREEN = 2# 定义 BLUE 数值为 3BLUE = 3col = Color.RED# 输出Color.REDprint(col)if col == Color.RED:# 判断 col 是否为 Color.RED，若是则输出 "Red"print("Red")# 输出 Color 枚举类的所有成员：[, , ]print(list(Color))# 判断col是否为Color类型print(isinstance(col, Color))# 输出 col 的类型：print(type(col))# 输出 col 的字符串表示形式：print(repr(col))# 通过 Color['RED'] 获取 Color.REDprint(Color['RED'])# 通过 Color(1) 获取 Color.REDprint(Color(1))Color.REDRed[, , ]TrueColor.REDColor.RED函数式API创建枚举类
# 引入枚举类型from enum import Enum# 使用functional API创建枚举类，定义三个枚举常量，分别为 RED, GREEN, BLUE，从数字2开始标号Color = Enum('Color', 'RED GREEN BLUE', start=2)# 或使用以下形式# Color = Enum('Color', [('RED', 1), ('BLUE', 2), ('GREEN', 3)])# 将 col 设为 GREENcol = Color.GREEN# 输出 col 的值：Color.GREENprint(col)# 比较 col 和 GREEN 是否相等，如果相等则输出 "Green"if col == Color.GREEN:print("Green")# 遍历枚举for color in Color:print(color)# 使用 name 和 value 属性获取枚举名称和值for color in Color:print(color.name, color.value)# 使用 __members__ 属性获取枚举类型 Color 中的所有枚举常量和名称for name, member in Color.__members__.items():print(name, member)Color.GREENGreenColor.REDColor.GREENColor.BLUERED 2GREEN 3BLUE 4RED Color.REDGREEN Color.GREENBLUE Color.BLUE自动赋值
# 导入枚举和自动赋值模块from enum import Enum, auto# 定义一个颜色的枚举类class Color(Enum):# 自动赋值，RED 的值为 1，后面的枚举类值依次加 1RED = auto()BLUE = auto()GREEN = auto()# 遍历 Color 枚举类，输出每个枚举值for color in Color:print(color)# 遍历 Color 枚举类，输出每个枚举值的名称和值for color in Color:print(color.name, color.value)Color.REDColor.BLUEColor.GREENRED 1BLUE 2GREEN 3异类值
from enum import Enum# 枚举成员的值类型可以不一样，但是非常不推荐这样做class UserResponse(Enum):YES = 1NO = "No"MAYBE = "Maybe"print(list(UserResponse))[, , ]混合枚举
from enum import Enum# 通过多重继承创建支持整数比较的枚举class Size(int, Enum):S = 1M = 2L = 3XL = 4print(Size.S > Size.M)False# 实现和上述代码一样的功能from enum import IntEnum# 通过IntEnum创建支持整数比较的枚举class Size(IntEnum):S = 1M = 2L = 3XL = 4print(Size.S > Size.M)False强制唯一值
from enum import Enum, unique# 当成员值不唯一，如果不添加unique装饰器，则代码运行成功，但只保留第一个出现具有相同值的成员。# 如果添加unique，则运行报错# @uniqueclass Color(Enum):RED = 1GREEN = 2BLUE = 3ORANGE = 3WHITE = 3for color in Color:print(color)Color.REDColor.GREENColor.BLUE基于位掩码的枚举类型enum.Flag
from enum import IntFlag, Flag, auto# 创建基于位掩码的枚举类型，注意成员值为2的幂，最好不要自定义# 如果想创建整数枚举继承IntFlag即可class Permissions(Flag):READ = auto()# 定义读权限WRITE = auto()# 定义写权限EXECUTE = auto()# 定义执行权限DELETE = auto()# 定义删除权限# 使用 name 和 value 属性获取枚举名称和值# 可以看到各个成员的值是2的幂for p in Permissions:print(p.name, p.value)# 使用枚举成员perms_rw = Permissions.READ | Permissions.WRITE# 用户拥有读和写权限# 可以看到perms_rw的值为3print(perms_rw.name,perms_rw.value)# 检查是否有某个权限# 使用&运算符来判断一个枚举值中是否包含某个特定的枚举值if perms_rw & Permissions.READ:# 如果用户拥有读权限print("用户拥有读权限")if perms_rw & Permissions.WRITE:# 如果用户拥有写权限print("用户拥有写权限")if perms_rw & Permissions.EXECUTE:# 如果用户拥有执行权限print("用户拥有执行权限") # 遍历所有权限for perm in Permissions: print(perm)READ 1WRITE 2EXECUTE 4DELETE 8None 3用户拥有读权限用户拥有写权限Permissions.READPermissions.WRITEPermissions.EXECUTEPermissions.DELETE2 参考
- enum
- Python enum
- 枚举(enum)介绍