在 Python 中将整数转换为枚举

在 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

在 Python 中将整数转换为枚举

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

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