在我的程序中使用Python将int转换为枚举？

Question

如何在python中int投射enum？

Answer 1

如果您想要灵活地在int枚举和枚举之间进行转换,您可以使用enum.IntEnum

import enum

class Color(enum.IntEnum):
    green = 1
    blue = 2
    red = 3
    yellow = 4

color_code = 4
# cast to enum
color = Color(color_code)

# cast back to int
color_code = int(color)

注意:如果您使用的是python <3.4,enum则已经向后移植,但您需要安装它,例如通过pip install enum

有关python中枚举的更多信息 - https://docs.python.org/3/library/enum.html