enum - 获取字符串转换的枚举值

Question

我定义了以下枚举

from enum import Enum


class D(Enum):
    x = 1
    y = 2


print(D.x)

现在打印出来的值是

D.x

相反，我希望打印枚举的值

1

可以做些什么来实现这个功能？

Answer 1

如果您要使用 f-string 打印值，那么您可以从 Enum 和 str 继承您的枚举。例如：

from enum import Enum


class D(str, Enum):
    x = 1
    y = 2


print(D.x)
print(f"{D.x}")

输出：

D.x
1

Answer 2

如果您想将枚举成员与Int 值进行比较，更好的方法是扩展IntEnum：

from enum import IntEnum


class D(IntEnum):
    x = 1
    y = 2


print(D.x)

通过这种方式，您可以将枚举值与整数进行比较，而无需显式调用.value：

>>> D.x == 1
True

有关更多信息，您可以查看 Python 文档的这一部分：Enum comparisons

Answer 3

使用最直接的 dunder 方法是 _repr_ 而不是 _str_，因为它还允许您在列表中以这种方式打印它。

class D(Enum):
  x = 1
  y = 2

  def __repr__(self):
      return self.value

print([D.x,D.y])
>>> [1, 2]

Answer 4

我使用以下实现了访问

class D(Enum):
    x = 1
    y = 2

    def __str__(self):
        return '%s' % self.value

现在我可以做到了

print(D.x) 得到1 作为结果。

如果您想打印x 而不是1，也可以使用self.name。

Answer 5

您正在打印枚举对象。如果您只想打印它，请使用 .value 属性：

print(D.x.value)

见Programmatic access to enumeration members and their attributes section：

如果您有一个枚举成员并且需要它的名称或值：

>>>
>>> member = Color.red
>>> member.name
'red'
>>> member.value
1

如果您只想提供自定义字符串表示，则可以向枚举添加 __str__ 方法：

class D(Enum):
    def __str__(self):
        return str(self.value)

    x = 1
    y = 2

演示：

>>> from enum import Enum
>>> class D(Enum):
...     def __str__(self):
...         return str(self.value)
...     x = 1
...     y = 2
... 
>>> D.x
<D.x: 1>
>>> print(D.x)
1