在 Python 中,enum
模块提供了一种用于创建枚举类型的方式。以下是使用 Python enum
的一些基本示例和说明:
1、创建枚举类型:
from enum import Enum
class Color(Enum):
RED = 1
GREEN = 2
BLUE = 3
2、使用枚举类型:
# 访问枚举成员
print(Color.RED) # 输出: Color.RED
# 获取枚举成员的值
print(Color.RED.value) # 输出: 1
3、遍历枚举类型:
# 遍历所有枚举成员
for color in Color:
print(color)
# 输出:
# Color.RED
# Color.GREEN
# Color.BLUE
4、比较枚举类型:
# 比较枚举成员
if Color.RED == Color.BLUE:
print("Colors are equal")
else:
print("Colors are not equal") # 输出: Colors are not equal
5、使用枚举类型作为函数参数:
def paint(color):
print(f"Painting in {color}")
paint(Color.BLUE) # 输出: Painting in Color.BLUE
6、将整数值转换为枚举成员:
value = 2
color = Color(value)
print(color) # 输出: Color.GREEN