如何使用 python 中的输入访问枚举成员

Question

我正在尝试通过输入访问 enum 变量的值。

这是我的代码：

class Animals(Enum):
    Dog = 1
    Cat = 2
    Cow = 3 

Choose = input('Choose an animal')

print(Animals.Choose.value)

这给了我一个错误可能是因为Animals 不包含Choose。

如何区分enum 中的成员和我的输入变量？

所以如果我输入Dog，它会给出1（Dog 变量的值）？

Answer 1

枚举具有内置的字符串访问和访问结果的各种方式。有关详细信息，请参阅Enum 的文档：

from enum import Enum

class Animal(Enum):
    Dog = 1
    Cat = 2
    Cow = 3 

choice = input('Choose an animal: ')

print(Animal[choice])
print(repr(Animal[choice]))
print(Animal[choice].value)

Choose an animal: Cat
Animal.Cat
<Animal.Cat: 2>
2

Answer 2

你可以试试getattr:

from enum import Enum
class Animals(Enum):
    Dog = 1
    Cat = 2
    Cow = 3 


Choose = input('Choose an animal')

print(getattr(Animals, Choose).value)

输出：

1

getattr代表“get attribute”，意思是它获取类中的变量，它的名字就是第二个参数。

Enums 已经内置了__getitem__ 方法，所以你可以直接用[] 括号来索引它，像这样：

print(Animals[Choose].value)

输出：

1