python - 枚举上的随机选择

Question

我想在 Enum 上使用 random.choice。

我试过了:

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

但此代码失败并出现 KeyError。如何选择 Enum 的随机成员？

Answer 1

安Enum不是一个序列，所以你不能把它传递给random.choice() ，它试图在 0 和 len(Foo) 之间选择一个索引.像字典一样，索引访问 Enum而是希望传入枚举 names，所以 Foo[<integer>]此处失败，返回 KeyError .

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

bar = random.choice(list(Foo))

这是因为 Enum does support iteration .

演示:

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