Question

我有一个枚举类，可以存储4个季节，并且希望能够转到下一个季节。当我到达结尾（秋天）时，我希望它回到第一个（冬天）。有没有一种简单的方法可以做到这一点，还是我最好只使用列表或其他结构。我在网上发现的一些答案似乎有些矫kill过正，所以我很好奇使用枚举是否值得。

from enum import Enum, auto

class Season(Enum):
    WINTER = auto()
    SPRING = auto()
    SUMMER = auto()
    FALL   = auto()

>>> season = Season.SUMMER
>>> season.next()
<Season.FALL: 4>
>>> season.next()
<Season.WINTER: 1>

Answer 1

由于Enum已经支持迭代协议，因此使用itertools.cycle非常简单：

>>> from itertools import cycle
>>> seasons = cycle(Season)
>>> next(seasons)
<Season.WINTER: 1>
>>> next(seasons)
<Season.SPRING: 2>
>>> next(seasons)
<Season.SUMMER: 3>
>>> next(seasons)
<Season.FALL: 4>
>>> next(seasons)
<Season.WINTER: 1>

有没有简单的方法可以循环遍历枚举？

1 个答案: