Python中使用枚举类

2023-01-31 00:01:15 python 枚举

开发中我们经常定义常量,

其实有更好的方法:为这样的枚举类型定义一个class类型,然后,每个常量都是class的一个唯一实例。
python中提供了Enum类来实现这个功能:

from enum import Enum

Numbers = Enum('Name', ('乔峰', '鸠摩智', '段誉', '阿紫', '虚竹','慕容复'))
for name, member in Numbers.__members__.items():
    print(name, '=>', member, ',', member.value)

print(Numbers.慕容复.value) # 值默认从1 开始
print(Numbers['虚竹']) # 用数组下标的形式访问属性
print(Numbers(2))  # 直接使用索引形式访问

输出:

乔峰 => Name.乔峰 , 1
鸠摩智 => Name.鸠摩智 , 2
段誉 => Name.段誉 , 3
阿紫 => Name.阿紫 , 4
虚竹 => Name.虚竹 , 5
慕容复 => Name.慕容复 , 6
6
Name.虚竹
Name.鸠摩智

上面我们就获得了Numbers类型的枚举类,可以直接使用: Numbers.乔峰 来引用一个常量,或者枚举它的所有成员。
value属性则是自动赋给成员的int常量,默认从1开始计数。
可见,我们既可以用成员名称引用枚举常量,又可以直接根据value的值获得枚举常量。我们再看一个例子:

from enum import Enum

class Gender(Enum):
    Male = "男"
    Female = "女"


class Person(object):
    def __init__(self, name, gender):
        self.name = name
        self.gender = gender


p = Person('乔峰', Gender.Male)
if p.gender == Gender.Male:  # 输出 纯爷们
    print('存爷们!') 
else:
    print('俺娘说了!')

 



相关文章