Python枚举类型Enum用法

雷子

发布于 2023-12-19 17:00:23

31900

代码可运行

文章被收录于专栏：雷子说测试开发雷子说测试开发

运行总次数：0

代码可运行

枚举类型可以看作是一种标签或是一系列常量的集合，通常用于表示某些特定的有限集合。

from enum import Enum

class Color(Enum):
    RED = 1
    GREEN = 2
    BLOCK = 3
print(Color.RED)
print(Color.RED.value)
print(Color.RED.name)

结果：

上面的代码创建了一个基本的枚举示例，与我们普通类不同，它继承 Enum 这个类。同时使用类获取属性的时候，输出的不是它的数值，而是会显示枚举项。这样更能增加代码的可读性。

Color.BLOCK=4

结果：

结果会报错

不存在 key 相同的枚举项（类变量）

在枚举类中存在相同的 key 值也是不允许的，比如下面代码有两个相同的 RED

from enum import Enum

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

结果：

枚举可以相同的值，但是值相同的各项 Key 都会当做别名

在枚举类中，可以有相同值，但是后面的枚举项都是第一个的别名

from enum import Enum

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

结果：

Color.RED

从结果中可以看到，输出的是 Color.RED 这个类型，而不是 Color.BLUE。如果你不想让枚举类中有相同的 value 值，Enum 也提供了的相应的处理方法

from enum import Enum
from enum import unique

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

print(Color.GREEN.value)

结果打印

这就是后续可以值必须唯一。

对枚举的遍历

from enum import Enum
from enum import unique

@unique
class Color(Enum):
    RED = 1
    GREEN = 2
    BLUE = 3
for  key in Color:
    print(key)

结果：

遍历所有的key，value：

from enum import Enum
from enum import unique

@unique
class Color(Enum):
    RED = 1
    GREEN = 2
    BLUE = 3
for key ,value  in Color.__members__.items():
    print(key,value)

结果：

枚举比较

from enum import Enum
from enum import unique

@unique
class Color(Enum):
    RED = 1
    GREEN = 2
    BLUE = 3
print(Color.RED == Color.GREEN)

结果：

除了通过继承 Enum 类的方法创建枚举类，还可以使用 Enum() 函数创建枚举类

from enum import Enum
#创建一个枚举类
Color = Enum("Color",('red','green','blue'))
#调用枚举成员的 3 种方式
print(Color.red)
print(Color['red'])
print(Color(1))
print(Color.green.name)
#调取枚举成员中的 value 和 name
print(Color.red.value)
print(Color.red.name)
#遍历枚举类中所有成员的 2 种方式
for color in Color:
    print(color)

结果：