Python从Enum类返回int

Question

目的是使用Enum类来允许一定数量的选项的可预测输入。

我希望得到整数作为输出，但如果我打印它，我会得到class.attrib。

如何获取整数？

from enum import Enum
class Stiffness(Enum):
FREE = 0
RIGID = 1
FLEXIBLE = 2
print(Stiffness.FLEXIBLE)

输出：Stiffness.FLEXIBLE

预期输出：2

Answer 1

只需打印value:

print(Stiffness.FLEXIBLE.value)
# 2

您可以查看枚举文档中有关枚举成员属性的部分。

Answer 2

您可以使用IntEnum而不是Enum。

这允许将Stiffness.FLEXIBLE与整数2互换使用，例如，您可以将其传递给需要整数的函数。