Python 中如何获取枚举(Enum)名称

使用枚举成员上的 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)