如何在 Python 中获取枚举值

Python 1年前 54

在 Python 中获取枚举中的值：

使用点符号访问特定的枚举成员，例如 Color.GREEN。
访问枚举成员的 value 属性，例如 Color.GREEN.value。

from enum import Enum


class Color(Enum):
    RED = 'stop'
    GREEN = 'go'
    YELLOW = 'get ready'


print(Color.GREEN.value)  # ?️ 'go'

print(Color['GREEN'].value)  # ?️ 'go'

print(Color.RED.value)  # ?️ 'stop'

print(Color['RED'].value)  # ?️ 'stop'

上述代码运行结果如下

如何在 Python 中获取枚举值

我们可以使用枚举成员的名称和值属性来获取枚举的名称和值。

from enum import Enum


class Color(Enum):
    RED = 'stop'
    GREEN = 'go'
    YELLOW = 'get ready'


print(Color.GREEN.value)  # ?️ 'go'

print(Color.GREEN.name)  # ?️ 'GREEN'

我们还可以使用方括号来访问枚举成员。

from enum import Enum


class Color(Enum):
    RED = 'stop'
    GREEN = 'go'
    YELLOW = 'get ready'


name = 'GREEN'

print(Color[name].value)  # ?️ 'go'

print(Color['GREEN'].value)  # ?️ 'go'

当我们提前不知道枚举成员的名称时，这很有用（因为它是从文件中读取或从 API 中获取的）。

我们可以使用列表推导来检查特定值是否在枚举中。

from enum import Enum


class Color(Enum):
    RED = 'stop'
    GREEN = 'go'
    YELLOW = 'get ready'


values = [member.value for member in Color]
print(values) # ?️ ['stop', 'go', 'get ready']

if 'go' in values:
    # ?️ this runs
    print('go is in values')

列表推导用于对每个元素执行一些操作，或者选择满足条件的元素子集。

in 运算符测试成员资格。例如，如果 x 是 l 的成员，则 x in l 计算结果为 True，否则计算结果为 False。

如果需要遍历枚举，可以使用简单的 for 循环。

from enum import Enum


class Color(Enum):
    RED = 'stop'
    GREEN = 'go'
    YELLOW = 'get ready'


for color in Color:
    print(color)
    print(color.name, color.value)