Python 中如何获取枚举（Enum）名称

Python 4月前 91

使用枚举成员上的 name 属性来获取名称，例如 Sizes.SMALL.name。如果只有对应的值，则将值传递给枚举类并访问 name 属性。

from enum import Enum


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


print(Sizes.SMALL.name)  # ?️ SMALL

# ?️ 按值访问枚举名称
print(Sizes(1).name)  # ?️ SMALL

print(Sizes['SMALL'].name)  # ?️ SMALL

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

from enum import Enum


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


print(Sizes.MEDIUM.name)  # ?️ MEDIUM
print(Sizes.MEDIUM.value)  # ?️ 2

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

from enum import Enum


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


name = 'LARGE'

print(Sizes[name].name)  # ?️ MEDIUM
print(Sizes[name].value)  # ?️ 2

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

如果我们只有与枚举成员对应的值，请将其传递给枚举类并访问 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

如果我们需要获取枚举中所有名称的列表，则可以使用列表推导。

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']

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

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

如果需要检查枚举中是否存在特定名称或特定值，可以使用 in 运算符。

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 'MEDIUM' in names:
    # ?️ this runs
    print('MEDIUM is in enum')

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)