如何从python枚举类中获取所有值？

Question

我使用Enum4库创建enum类，如下所示：

class Color(Enum):
    RED = 1
    BLUE = 2

我想在某处打印[1,2]作为列表。如何实现这一点？

Answer 1

您可以使用IntEnum：

from enum import IntEnum

class Color(IntEnum):
   RED = 1
   BLUE = 2


print(int(Color.RED))   # prints 1

要获取整数列表：

enum_list = list(map(int, Color))
print(enum_list) # prints [1, 2]

Answer 2

根据@Jeff的回答，重构为使用类方法，以便可以对任何枚举重用相同的代码：

from enum import Enum

class ExtendedEnum(Enum):

    @classmethod
    def list(cls):
        return list(map(lambda c: c.value, cls))

class OperationType(ExtendedEnum):
    CREATE = 'CREATE'
    STATUS = 'STATUS'
    EXPAND = 'EXPAND'
    DELETE = 'DELETE'

print(OperationType.list())

生产：

['CREATE', 'STATUS', 'EXPAND', 'DELETE']

Answer 3

您可以执行以下操作：

[e.value for e in Color]