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

enum库

news 来源：原创 2024/9/29 11:41:21

Python `enum` 模块教程

enum 是 Python 3.4 引入的一个模块，用于定义枚举类型。枚举类型是一种特殊的数据类型，由一组命名的值组成，这些值称为枚举成员。使用 enum 可以提高代码的可读性和可维护性，特别是在处理一组相关的常量值时。

官方文档链接

Python Enum官方文档

枚举类型的定义和使用

定义枚举类型

你可以使用 Enum 类来定义枚举类型。每个枚举成员由名称和值组成：

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

访问枚举成员

枚举成员可以通过名称或值来访问：

# 通过名称访问
print(Color.RED)
print(Color['RED'])# 通过值访问
print(Color(1))

枚举成员的属性

每个枚举成员有两个主要属性：name 和 value：

print(Color.RED.name)   # 输出: RED
print(Color.RED.value)  # 输出: 1

枚举的迭代和比较

迭代枚举成员

你可以使用 for 循环来迭代枚举成员：

for color in Color:print(color)

比较枚举成员

枚举成员可以使用 == 和 is 进行比较：

print(Color.RED == Color.RED)  # 输出: True
print(Color.RED is Color.RED)  # 输出: Trueprint(Color.RED == Color.BLUE)  # 输出: False
print(Color.RED is Color.BLUE)  # 输出: False

枚举的高级功能

自定义枚举成员值

枚举成员的值可以是任何数据类型，不限于整数：

class Shape(Enum):CIRCLE = 'circle'SQUARE = 'square'TRIANGLE = 'triangle'

自动赋值枚举成员

使用 auto() 可以自动为枚举成员赋值：

from enum import autoclass Animal(Enum):DOG = auto()CAT = auto()BIRD = auto()print(list(Animal))  # 输出: [<Animal.DOG: 1>, <Animal.CAT: 2>, <Animal.BIRD: 3>]

枚举成员的别名

同一个值可以有多个名称，这些名称称为别名：

class Status(Enum):SUCCESS = 0FAILURE = 1ERROR = 1print(Status.SUCCESS)  # 输出: Status.SUCCESS
print(Status.FAILURE)  # 输出: Status.FAILURE
print(Status.ERROR)    # 输出: Status.FAILURE

扩展枚举

可以通过继承 Enum 类来扩展枚举：

class ExtendedColor(Color):YELLOW = 4BLACK = 5

自定义方法

枚举类可以定义自定义方法：

class Planet(Enum):MERCURY = 1VENUS = 2EARTH = 3def describe(self):return f'Planet {self.name} is number {self.value} from the sun.'print(Planet.EARTH.describe())  # 输出: Planet EARTH is number 3 from the sun.

使用示例

以下是一个完整的示例，展示了如何定义和使用枚举类型：

from enum import Enum, autoclass Color(Enum):RED = auto()GREEN = auto()BLUE = auto()class Shape(Enum):CIRCLE = 'circle'SQUARE = 'square'TRIANGLE = 'triangle'def describe_color(color):if color == Color.RED:return "The color is red."elif color == Color.GREEN:return "The color is green."elif color == Color.BLUE:return "The color is blue."else:return "Unknown color."def describe_shape(shape):return f'The shape is a {shape.value}.'print(describe_color(Color.RED))         # 输出: The color is red.
print(describe_color(Color.GREEN))       # 输出: The color is green.
print(describe_shape(Shape.CIRCLE))      # 输出: The shape is a circle.
print(describe_shape(Shape.SQUARE))      # 输出: The shape is a square.