迹忆客 专注技术分享

要检查 Python 中的枚举中是否存在值：

1. 使用列表推导获取所有枚举值的列表。
2. 使用 in 运算符检查值是否存在于列表中。
3. 如果值在列表中， in 运算符将返回 True。

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 enum values')
else:
    print('2 is NOT in enum values')

print(100 in values) # 👉️ False

我们使用列表推导来获取枚举值的列表。

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

最后一步是使用 in 运算符检查特定值是否在列表中。

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

如果我们需要检查一个值是否不在枚举中，请用 not 否定条件。

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 100 not in values:
    # 👇️ this runs
    print('100 is NOT in enum values')

如果我们需要检查枚举中是否存在特定名称，则可以使用相同的方法。

from enum import Enum


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


names = [member.name for member in Sizes]

print(names)  # 👉️ ['SMALL', 'MEDIUM', 'LARGE']

if 'SMALL' in names:
    # 👇️ this runs
    print('SMALL is in enum names')

我们没有访问 value 属性，而是使用 name 属性来获取枚举名称的列表。