Python中使用枚举类

发布时间:2019-05-23 20:50:40编辑:auto阅读(2041)

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

    其实有更好的方法:为这样的枚举类型定义一个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('俺娘说了!')

     



关键字