python 3.4:在Enum上的random.choice

Question

我想在Enum上使用random.choice.

我试过了 :

class Foo(Enum):
    a = 0
    b = 1
    c = 2
bar = random.choice(Foo)

但是这段代码不起作用,我怎么能这样做？

Answer 1

一个Enum不是一个序列,所以你不能将它传递给它random.choice(),它试图选择0和之间的索引len(Foo).像字典一样,索引访问一个Enum代替期望枚举名称传入,所以Foo[<integer>]这里失败了KeyError.

您可以先将其强制转换为列表:

bar = random.choice(list(Foo))

这工作,因为Enum 不支持迭代.

演示:

>>> from enum import Enum
>>> import random
>>> class Foo(Enum):
...     a = 0
...     b = 1
...     c = 2
... 
>>> random.choice(list(Foo))
<Foo.a: 0>