Articles
Python简介
Python中的枚举

Python中的枚举

本文讲解了Python中的枚举。

YouTube Video

Python中的枚举

Python中的enum是一个用于定义常量的特殊类，它可以提高代码可读性并帮助防止编码错误。

什么是Enum？

Enum（枚举的缩写）是一个用于定义一组命名常量的类。它通常用于将相关的常量分组在一起。例如，它适用于表示颜色、星期几或方向等。

为什么使用Enum？

有几个理由使用 Enum。

  • 提高可读性：使用命名常量使代码更加直观。
  • 防止错误：它可以防止常量值的更改或误用。
  • 分组：它将相关常量组织到一个类中，表示结构化数据。

Enum的基本用法

使用enum时，需要导入enum模块并定义一个类。该类应继承自Enum并定义你希望作为常量的值。

Enum的基本示例 

 1from enum import Enum
 2
 3class Color(Enum):
 4    RED = 1
 5    GREEN = 2
 6    BLUE = 3
 7
 8# Usage of Enum
 9print(Color.RED)        # Color.RED
10print(Color.RED.name)   # RED
11print(Color.RED.value)  # 1

在此示例中，我们定义了一个名为Color的Enum类，其中包含三个值。每个名称可以通过name属性访问，其关联值可以通过value属性访问。

枚举的比较

Enum成员可以通过名称或值进行比较。你可以使用is运算符或==运算符检查身份和相等性。

比较示例 

 1from enum import Enum
 2
 3class Direction(Enum):
 4    NORTH = 1
 5    SOUTH = 2
 6    EAST = 3
 7    WEST = 4
 8
 9# Equality comparison
10print(Direction.NORTH == Direction.SOUTH)  # False
11print(Direction.NORTH == Direction.NORTH)  # True
12
13# Identity comparison using is
14print(Direction.NORTH is Direction.SOUTH)  # False
15print(Direction.NORTH is Direction.NORTH)  # True

这里演示了使用==和is进行Enum比较的方式。Enum被设计用来正确地比较具有相同名称和值的对象。

使用auto()的自动值分配

如果您希望自动分配Enum成员的值，可以使用enum.auto()。auto()使Python能够自动分配值，省去了开发人员手动指定的需求。

auto()示例 

 1from enum import Enum, auto
 2
 3class Animal(Enum):
 4    DOG = auto()
 5    CAT = auto()
 6    MOUSE = auto()
 7
 8# Check the values of Enum
 9print(Animal.DOG.value)    # 1
10print(Animal.CAT.value)    # 2
11print(Animal.MOUSE.value)  # 3

在此示例中，使用auto()省去了明确指定值的需求，使代码更加简洁。

为Enum成员分配多个值

Enum成员还可以包含多个值，例如元组。这允许每个成员持有额外的相关信息。

使用元组的示例 

 1from enum import Enum
 2
 3class Status(Enum):
 4    ACTIVE = (1, 'Active user')
 5    INACTIVE = (2, 'Inactive user')
 6    SUSPENDED = (3, 'Suspended user')
 7
 8# Accessing Enum members
 9print(Status.ACTIVE.name)    # ACTIVE
10print(Status.ACTIVE.value)   # (1, 'Active user')
11print(Status.ACTIVE.value[1])  # Active user

在此示例中，Status类的每个成员都有两个值，分别表示状态和描述。当您希望将多个信息与Enum成员关联时，这非常有用。

将Enum用作标志

Python的enum包括一个Flag类，可以像位标志一样使用。使用Flag允许您组合管理多个状态。

Flag示例 

 1from enum import Flag, auto
 2
 3class Permission(Flag):
 4    READ = auto()
 5    WRITE = auto()
 6    EXECUTE = auto()
 7
 8# Combining flags
 9permission = Permission.READ | Permission.WRITE
10print(permission)                 # Permission.READ|WRITE
11print(Permission.EXECUTE in permission)  # False

在此示例中，使用位运算组合多个权限。使用Flag使您能够简洁地管理多个状态。

使用Enum时需要注意的事项

Enum的不可变性

Enum成员是不可变的。一旦定义，它们的名称或值就不能被更改。这是一项重要特性，用于作为常量保持一致性。

1from enum import Enum
2
3class Days(Enum):
4    MONDAY = 1
5    TUESDAY = 2
6
7# Example of immutability
8# Days.MONDAY = 3  # AttributeError: Cannot reassign members.

禁止重复成员

Enum成员必须具有唯一的名称和值。即使允许重复的值，Python也会自动优先选取第一个成员。

总结

Python的enum模块是一个非常有用的工具，可用于将常量分组、提高可读性和维护代码一致性。从基本使用到通过auto()进行自动赋值，以及用作位标志，都有多种可能的用法。

正确理解和掌握enum可以提高代码质量并有助于防止错误。

您可以在我们的YouTube频道上使用Visual Studio Code跟随上述文章进行学习。 请也查看我们的YouTube频道。

YouTube Video

Python 中的类Python中的迭代器