在 Python 中将枚举转换为整数

Python 4月前 6

使用 enum 模块中的 IntEnum 类将枚举转换为 Python 中的整数。如果确切的值不重要，我们可以使用 auto() 类。要获取枚举成员的值，请使用该成员的 value 属性。

from enum import IntEnum


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


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

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

在 Python 中将枚举转换为整数

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

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

如果枚举成员的确切值不重要，我们也可以使用 auto() 类。

from enum import IntEnum, auto


class Sizes(IntEnum):
    SMALL = auto()
    MEDIUM = auto()
    LARGE = auto()


# ?️ [<Sizes.SMALL: 1>, <Sizes.MEDIUM: 2>, <Sizes.LARGE: 3>]
print(list(Sizes))

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

默认情况下，使用 auto 类时的初始值从 1 开始。

如果必须将枚举成员与整数进行比较，则从 IntEnum 类扩展很有用。

from enum import IntEnum


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


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

或者，我们可以使用 enum 模块中更通用的 Enum 类。

from enum import Enum


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


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

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

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

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

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