在字符串转换时获取枚举的值

问题描述

我定义了以下枚举

from enum import Enum


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


print(D.x)

现在打印值为

D.x

相反，我希望枚举的值为print

1

如何实现此功能？

---

解决方案

您正在打印枚举对象。如果您只想打印：

，请使用.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