迹忆客 专注技术分享

在 Python 中使用括号将整数转换为枚举，例如 print(Sizes(1))。 我们可以将枚举成员的整数值作为参数传递给类，它将返回枚举成员。

from enum import Enum


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


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

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

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

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

如果必须将枚举的成员与整数进行比较，也可以从 IntEnum 类扩展。

from enum import IntEnum


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


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

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

print(Sizes.SMALL == 1)  # 👉️ True
print(Sizes.MEDIUM == 2)  # 👉️ True

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

from enum import Enum


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


name = 'MEDIUM'

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

print(Sizes['SMALL'].value)  # 👉️ 1

这是一个接受用户输入并使用它来访问枚举成员的示例。

from enum import Enum


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


value = input('Pick size - 1, 2 or 3: ')

print(value)  # 👉️ 2

print(Sizes(int(value)))  # 👉️ Sizes.MEDIUM

print(Sizes(int(value)).name)  # 👉️ MEDIUM
print(Sizes(int(value)).value)  # 👉️ 2

如果需要遍历枚举，可以使用简单的 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)

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

from enum import Enum


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


# 👇️ check enum member type
print(type(Sizes.MEDIUM))  # 👉️ <enum 'Sizes'>

# 👇️ check if member belongs to Enum
print(isinstance(Sizes.MEDIUM, Sizes))  # 👉️ True

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

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

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