从多个值中获取枚举名称 python

Question

我正在尝试获取给定多个值之一的枚举的名称：

class DType(Enum):
    float32 = ["f", 8]
    double64 = ["d", 9]

当我尝试获取一个给出名称的值时，它起作用了：

print DType["float32"].value[1]  # prints 8
print DType["float32"].value[0]  # prints f

但是当我尝试从给定值中获取名称时，只会出现错误：

print DataType(8).name
print DataType("f").name

raise ValueError("%s is not a valid %s" % (value, cls.name))

ValueError: 8 is not a valid DataType

ValueError: f is not a valid DataType

有没有办法做到这一点？还是我使用了错误的数据结构？

Answer 1

最简单的方法是使用 aenum 库 ¹，它看起来像这样：

from aenum import MultiValueEnum

class DType(MultiValueEnum):
    float32 = "f", 8
    double64 = "d", 9

正在使用：

>>> DType("f")
<DType.float32: 'f'>

>>> DType(9)
<DType.double64: 'd'>

如您所见，列出的第一个值是规范值，并显示在 repr().

如果你想显示所有可能的值，或者需要使用标准库 Enum (Python 3.4+)，那么 answer found here 是你的基础想要（也将与 aenum 一起使用）：

class DType(Enum):
    float32 = "f", 8
    double64 = "d", 9

    def __new__(cls, *values):
        obj = object.__new__(cls)
        # first value is canonical value
        obj._value_ = values[0]
        for other_value in values[1:]:
            cls._value2member_map_[other_value] = obj
        obj._all_values = values
        return obj

    def __repr__(self):
        return '<%s.%s: %s>' % (
                self.__class__.__name__,
                self._name_,
                ', '.join([repr(v) for v in self._all_values]),
                )

正在使用：

>>> DType("f")
<DType.float32: 'f', 8>

>>> Dtype(9)
<DType.float32: 'f', 9>

¹ 披露：我是 Python stdlib Enum, the enum34 backport, and the Advanced Enumeration (aenum) 库的作者。

从多个值中获取枚举名称 python

Get Enum name from multiple values python

python

enums

multiple-value