主页 / coding / 问题 / 78169351
Accepted
Jan
Asked: 2024-03-16 03:46:28 +0800 CST

返回两个枚举状态

  • 772

假设，我有

class State(Enum):
    TAKEPROFIT  = 1
    STOPPEDOUT  = 2
    WINNER      = 3
    LOSER       = 4

如何返回egState.STOPPEDOUT和的组合State.LOSER？
似乎|不支持：

return State.STOPPEDOUT | State.LOSER

投掷

TypeError: unsupported operand type(s) for |: 'State' and 'State'
python

3 个回答

  1. Best Answer

    使用 anIntEnum代替Enum:

    from enum import IntEnum


class State(IntEnum):
    TAKEPROFIT = 1
    STOPPEDOUT = 2
    WINNER = 3
    LOSER = 4

    An 的IntEnum行为类似于int，因此：

    >>> State.STOPPEDOUT | State.LOSER
6
    • 4

  2. 虽然 @larsks 的答案是正确的，但“更”正确的方法是继承IntFlag（auto()生成 2 的幂；如果您选择，可以使用自己的值）：

    from enum import IntFlag, auto

class State(IntFlag):
  TAKEPROFIT = auto()
  STOPPEDOUT = auto()
  WINNER = auto()
  LOSER = auto()

    |，&并且^操作会产生同一类的实例：

    >>> State.STOPPEDOUT | State.LOSER
<State.STOPPEDOUT|LOSER: 10>
>>> State.STOPPEDOUT & State.LOSER
<State: 0>
>>> State.STOPPEDOUT ^ (State.LOSER | State.STOPPEDOUT | State.TAKEPROFIT)
<State.TAKEPROFIT|LOSER: 9>

    in还支持以下操作：

    >>> State.STOPPEDOUT in (State.STOPPEDOUT | State.LOSER)
True

    IntFlagint是and的子类Flag（它又是 的子类Enum），因此它的实例可以在任何int需要 an 的地方使用：

    >>> State.WINNER + 9
13

    但是，如果 anIntFlag被反转，它将返回其余标志的并集：

    >>> ~State.TAKEPROFIT
<State.STOPPEDOUT|WINNER|LOSER: 14>
    • 2

  3. 枚举有一个 value 属性：

    class State(Enum):
    TAKEPROFIT  = 1
    STOPPEDOUT  = 2
    WINNER      = 3
    LOSER       = 4

State.STOPPEDOUT.value | State.LOSER.value # 6
    • 0

相关问题