当前位置：首页 > news >正文

Python教程：python枚举类定义和使用

news 2025/7/28 3:05:39

在Python中，枚举类（Enum）用于表示一组常量，使代码更加清晰和易于维护。枚举类通过enum模块定义。以下是如何定义和使用枚举类的详细步骤。

定义枚举类

首先，你需要导入enum模块，然后定义一个枚举类。枚举类中的每个成员都是该类的实例，并且每个成员都有一个唯一的名称和值。

from enum import Enum  class Color(Enum):  RED = 1  GREEN = 2  BLUE = 3

在这个例子中，Color是一个枚举类，它有三个成员：RED、GREEN和BLUE，每个成员都有一个对应的整数值。

使用枚举类

你可以像使用普通类一样使用枚举类。枚举类的成员可以通过类名来访问，并且每个成员都有name和value属性。

# 访问枚举成员  
print(Color.RED)          # 输出: Color.RED  
print(Color.RED.name)     # 输出: RED  
print(Color.RED.value)    # 输出: 1  # 遍历枚举成员  
for color in Color:  print(color.name, color.value)  
# 输出:  
# RED 1  
# GREEN 2  
# BLUE 3

枚举的比较和身份

枚举成员可以进行比较和身份检查。

# 比较枚举成员  
print(Color.RED == Color.RED)   # 输出: True  
print(Color.RED == Color.GREEN) # 输出: False  # 身份检查  
print(Color.RED is Color(1))    # 输出: True

枚举成员的唯一性

枚举成员的值在枚举类中必须是唯一的。如果尝试定义两个具有相同值的成员，将会引发ValueError。

class InvalidColor(Enum):  RED = 1  ANOTHER_RED = 1  # 这将引发 ValueError

自定义枚举值

枚举成员的值不必是整数，它们可以是任何不可变类型（例如字符串、元组等）。

class Status(Enum):  PENDING = 'pending'  APPROVED = 'approved'  REJECTED = 'rejected'  print(Status.PENDING.value)  # 输出: pending

枚举的自动方法和属性

枚举类有一些有用的方法和属性：

members：返回包含所有成员名称和成员的字典。
iter()：允许枚举类被迭代。

# 获取所有成员  
members = Color.__members__  
print(members)  # 输出: {'RED': <Color.RED: 1>, 'GREEN': <Color.GREEN: 2>, 'BLUE': <Color.BLUE: 3>}  # 检查成员是否存在  
print('RED' in Color.__members__)  # 输出: True  
print('YELLOW' in Color.__members__)  # 输出: False

示例：使用枚举类处理状态

假设你有一个订单处理系统，可以使用枚举类来表示订单的状态。

class OrderStatus(Enum):  CREATED = 1  IN_PROGRESS = 2  COMPLETED = 3  CANCELLED = 4  # 示例订单  
order = {  'id': 123,  'status': OrderStatus.CREATED  
}  # 检查订单状态  
if order['status'] == OrderStatus.CREATED:  print("Order has been created.")  
elif order['status'] == OrderStatus.COMPLETED:  print("Order has been completed.")  
else:  print("Order is in another status.")