Enum在Python|极客教程

Enum在Python

什么是Enum

Enum（枚举）是Python中一个非常有用的数据类型，用于定义有限个数的常量。Enum在Python中由enum模块提供支持，它允许我们为一组唯一标识符分配符号名称。通过使用Enum，我们可以避免使用常量来表示不同的状态或选项，从而使代码更加可读和易于维护。

如何创建Enum

我们可以通过继承Enum类来创建枚举。下面是一个简单的示例代码：

from enum import Enum

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

在这个示例中，我们创建了一个名为Color的枚举，它包含三种颜色：红色，绿色和蓝色。我们可以通过以下方式访问枚举值：

print(Color.RED)

运行结果：

Color.RED

Enum的属性和方法

Enum类提供了一些有用的属性和方法来操作枚举。其中一些常用的属性和方法包括：

name：返回枚举成员的名称
value：返回枚举成员的值
__members__：返回枚举所有成员的字典

下面是示例代码：

from enum import Enum

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

print(Color.RED.name)
print(Color.RED.value)
print(Color.__members__)

运行结果：

RED
1
{'RED': <Color.RED: 1>, 'GREEN': <Color.GREEN: 2>, 'BLUE': <Color.BLUE: 3>}

枚举比较

枚举类实现了比较操作符，可以直接使用==和!=来比较枚举值。下面是示例代码：

from enum import Enum

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

print(Color.RED == Color.RED)
print(Color.RED != Color.GREEN)

运行结果：

True
True

使用Enum实现单例模式

我们可以使用枚举来实现单例模式，确保一个类只有一个实例。下面是一个示例代码：

from enum import Enum

class Singleton(Enum):
    INSTANCE = 1

class SingletonClass:
    def __new__(cls):
        if not hasattr(cls, 'instance'):
            cls.instance = super().__new__(cls)
        return cls.instance

s1 = SingletonClass()
s2 = SingletonClass()

print(s1 is s2)

运行结果：

True

Enum的高级用法

除了简单的枚举之外，Enum还支持一些高级用法，例如自定义枚举值、别名、有序枚举等。下面是一些示例代码：

自定义枚举值

from enum import Enum, unique

@unique
class Color(Enum):
    RED = 'red'
    GREEN = 'green'
    BLUE = 'blue'

print(Color.RED.value)

运行结果：

red

别名

from enum import Enum

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

Color.RED_ALIAS = Color.RED
print(Color.RED is Color.RED_ALIAS)

运行结果：

True

有序枚举

from enum import Enum

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

from enum import Enum

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

@unique
class Size(Enum):
    SMALL = 1
    MEDIUM = 2
    LARGE = 3

print(Size.SMALL < Size.MEDIUM)

运行结果：