欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

Python中使用枚举类

程序员文章站 2022-04-26 16:54:30
开发中我们经常定义常量, 其实有更好的方法:为这样的枚举类型定义一个class类型,然后,每个常量都是class的一个唯一实例。Python中提供了Enum类来实现这个功能: 输出: 上面我们就获得了Numbers类型的枚举类,可以直接使用: Numbers.乔峰 来引用一个常量,或者枚举它的所有成 ......

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

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