迹忆客 专注技术分享

在 Python 中要按值获取枚举名称，请将值传递给枚举类并访问名称属性，例如 Sizes(1).name。 当值被传递给类时，我们可以访问相应的枚举成员，我们可以在其上访问 name 属性。

from enum import Enum

class Sizes(Enum):
    SMALL = 1
    MEDIUM = 2
    LARGE = 3


print(Sizes(1).name)  # 👉️ SMALL
print(Sizes(2).name)  # 👉️ MEDIUM
print(Sizes(3).name)  # 👉️ LARGE

Sizes(1) 语法允许我们将整数传递给类并获取相应的枚举成员。

from enum import Enum


class Sizes(Enum):
    SMALL = 1
    MEDIUM = 2
    LARGE = 3


value = 1

print(Sizes(value))  # 👉️ Sizes.SMALL

print(Sizes(value).name)  # 👉️ SMALL

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

如果我们需要获取枚举名称或值的列表，请使用列表推导。

from enum import Enum


class Sizes(Enum):
    SMALL = 1
    MEDIUM = 2
    LARGE = 3


values = [member.value for member in Sizes]
print(values)  # 👉️ [1, 2, 3]

names = [member.name for member in Sizes]
print(names)  # 👉️ ['SMALL', 'MEDIUM', 'LARGE']

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

如果我们需要获取包含每个枚举成员的名称和值的元组列表，则可以使用相同的方法。

from enum import Enum


class Sizes(Enum):
    SMALL = 1
    MEDIUM = 2
    LARGE = 3


result = [(member.name, member.value) for member in Sizes]

# 👇️ [('SMALL', 1), ('MEDIUM', 2), ('LARGE', 3)]
print(result)

每个元组中的第一个元素是名称，第二个元素是枚举成员的值。

如果需要检查值是否在枚举中，请使用 in 运算符。

from enum import Enum


class Sizes(Enum):
    SMALL = 1
    MEDIUM = 2
    LARGE = 3


values = [member.value for member in Sizes]
print(values)  # 👉️ [1, 2, 3]

if 2 in values:
    # 👇️ this runs
    print('2 is in values')

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

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

from enum import Enum


class Sizes(Enum):
    SMALL = 1
    MEDIUM = 2
    LARGE = 3


for size in Sizes:
    print(size)
    print(size.name, size.value)