Python中的枚举类型示例介绍

2023-09-01 18:11:05 69

起步

Python的原生类型中并不包含枚举类型。为了提供更好的解决方案，Python通过PEP435在3.4版本中添加了enum标准库。

枚举类型可以看作是一种标签或是一系列常量的集合，通常用于表示某些特定的有限集合，例如星期、月份、状态等。在没有专门提供枚举类型的时候我们是怎么做呢，一般就通过字典或类来实现：

Color={
'RED':1,
'GREEN':2,
'BLUE':3,
}

classColor:
RED=1
GREEN=2
BLUE=3

这种来实现枚举如果小心翼翼地使用当然没什么问题，毕竟是一种妥协的解决方案。它的隐患在于可以被修改。

使用Enum

更好的方式是使用标准库提供的Enum类型，官方库值得信赖。3.4之前的版本也可以通过pipinstallenum下载支持的库。简单的示例：

fromenumimportEnum
classColor(Enum):
red=1
green=2
blue=3

枚举成员有值（默认可重复），枚举成员具有友好的字符串表示：

>>>print(Color.red)
Color.red
>>>print(repr(Color.red))

>>>type(Color.red)

>>>isinstance(Color.green,Color)
True

枚举类型不可实例化，不可更改。

定义枚举

定义枚举时，成员名不允许重复

classColor(Enum):
red=1
green=2
red=3#TypeError:Attemptedtoreusekey:'red'

成员值允许相同，第二个成员的名称被视作第一个成员的别名

classColor(Enum):
red=1
green=2
blue=1

print(Color.red)#Color.red
print(Color.blue)#Color.red
print(Color.redisColor.blue)#True
print(Color(1))#Color.red在通过值获取枚举成员时，只能获取到第一个成员

若要不能定义相同的成员值，可以通过unique装饰

fromenumimportEnum,unique
@unique
classColor(Enum):
red=1
green=2
blue=1#ValueError:duplicatevaluesfoundin:blue->red

枚举取值

可以通过成员名来获取成员也可以通过成员值来获取成员:

print(Color['red'])#Color.red通过成员名来获取成员

print(Color(1))#Color.red通过成员值来获取成员

每个成员都有名称属性和值属性：

member=Color.red
print(member.name)#red
print(member.value)#1

支持迭代的方式遍历成员，按定义的顺序，如果有值重复的成员，只获取重复的第一个成员：

forcolorinColor:
print(color)

特殊属性__members__是一个将名称映射到成员的有序字典，也可以通过它来完成遍历：

forcolorinColor.__members__.items():
print(color)#('red',)

枚举比较

枚举的成员可以通过is同一性比较或通过==等值比较：

Color.redisColor.red
Color.redisnotColor.blue

Color.blue==Color.red
Color.blue!=Color.red

枚举成员不能进行大小比较：

Color.red
扩展枚举IntEnum


IntEnum是Enum的扩展，不同类型的整数枚举也可以相互比较：
fromenumimportIntEnum
classShape(IntEnum):
circle=1
square=2

classRequest(IntEnum):
post=1
get=2

print(Shape.circle==1)#True
print(Shape.circle<3)#True
print(Shape.circle==Request.post)#True
print(Shape.circle>=Request.post)#True
总结


enum模块功能很明确，用法也简单，其实现的方式也值得学习，有机会的话可以看看它的源码。
好了，以上就是这篇文章的全部内容了，希望本文的内容对大家的学习或者工作具有一定的参考学习价值，如果有疑问大家可以留言交流，谢谢大家对毛票票的支持。

Python中的枚举类型示例介绍

热门推荐

随机推荐