Python 的枚举类型
2024-09-03 05:20:30
起步
Python 的原生类型中并不包含枚举类型。为了提供更好的解决方案,Python 通过 PEP 435 在 3.4 版本中添加了 enum
标准库。
枚举类型可以看作是一种标签或是一系列常量的集合,通常用于表示某些特定的有限集合,例如星期、月份、状态等。在没有专门提供枚举类型的时候我们是怎么做呢,一般就通过字典或类来实现:
Color = {
'RED' : 1,
'GREEN': 2,
'BLUE' : 3,
}
class Color:
RED = 1
GREEN = 2
BLUE = 3
这种来实现枚举如果小心翼翼地使用当然没什么问题,毕竟是一种妥协的解决方案。它的隐患在于可以被修改。
使用 Enum
更好的方式是使用标准库提供的 Enum
类型,官方库值得信赖。3.4 之前的版本也可以通过 pip install enum
下载支持的库。简单的示例:
from enum import Enum
class Color(Enum):
red = 1
green = 2
blue = 3
枚举成员有值(默认可重复),枚举成员具有友好的字符串表示:
>>> print(Color.red)
Color.red
>>> print(repr(Color.red))
<Color.red: 1>
>>> type(Color.red)
<Enum 'Color'>
>>> isinstance(Color.green, Color)
True
枚举类型不可实例化,不可更改。
定义枚举
定义枚举时,成员名不允许重复
class Color(Enum):
red = 1
green = 2
red = 3 # TypeError: Attempted to reuse key: 'red'
成员值允许相同,第二个成员的名称被视作第一个成员的别名
class Color(Enum):
red = 1
green = 2
blue = 1
print(Color.red) # Color.red
print(Color.blue) # Color.red
print(Color.red is Color.blue)# True
print(Color(1)) # Color.red 在通过值获取枚举成员时,只能获取到第一个成员
若要不能定义相同的成员值,可以通过 unique 装饰
from enum import Enum, unique
@unique
class Color(Enum):
red = 1
green = 2
blue = 1 # ValueError: duplicate values found in <enum 'Color'>: blue -> red
枚举取值
可以通过成员名来获取成员也可以通过成员值来获取成员:
print(Color['red']) # Color.red 通过成员名来获取成员
print(Color(1)) # Color.red 通过成员值来获取成员
每个成员都有名称属性和值属性:
member = Color.red
print(member.name) # red
print(member.value) # 1
支持迭代的方式遍历成员,按定义的顺序,如果有值重复的成员,只获取重复的第一个成员:
for color in Color:
print(color)
特殊属性 __members__
是一个将名称映射到成员的有序字典,也可以通过它来完成遍历:
for color in Color.__members__.items():
print(color) # ('red', <Color.red: 1>)
枚举比较
枚举的成员可以通过 is
同一性比较或通过 ==
等值比较:
Color.red is Color.red
Color.red is not Color.blue
Color.blue == Color.red
Color.blue != Color.red
枚举成员不能进行大小比较:
Color.red < Color.blue # TypeError: unorderable types: Color() < Color()
扩展枚举 IntEnum
IntEnum
是 Enum
的扩展,不同类型的整数枚举也可以相互比较:
from enum import IntEnum
class Shape(IntEnum):
circle = 1
square = 2
class Request(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
模块功能很明确,用法也简单,其实现的方式也值得学习,有机会的话可以看看它的源码。
来源:https://segmentfault.com/a/1190000017327003
最新文章
- [WPF系列]Button 自定义
- 快手4.0 (KSCAD)
- SQL语句删除所有表
- Sharepoint学习笔记—习题系列--70-573习题解析 --索引目录
- CSS初学
- mongodb 3.2 分片部署步骤
- Spark 3000门徒第一课随笔
- vector C++ 详细用法
- MFC最大化显示任务栏
- js修改input的type属性问题
- Http,Https (SSL)的Url绝对路径,相对路径解决方案Security Switch 4.2 中文帮助文档 分类: ASP.NET 2014-10-28 14:09 177人阅读 评论(1) 收藏
- JDBC批量操作
- 最后的配置部分:LNMP+Tomcat
- html&;&;css 基础知识笔记
- CTF入门指南(0基础)
- 【Qt官方MQTT库的使用,附一个MqttClient例子】
- 读Java并发编程实践中,向已有线程安全类添加功能--客户端加锁实现示例
- HDU 2389 Rain on your Parade 最大匹配(模板题)【HK算法】
- linux内存源码分析 - 伙伴系统(初始化和申请页框)
- LOJ #2802. 「CCC 2018」平衡树(整除分块 + dp)
热门文章
- (1)hello world
- Tyvj——P1952 Easy
- Xamarin.Forms的相对布局RelativeLayout
- jenkins发布普通项目、配置自动上线自动部署
- luogu P2746 [USACO5.3]校园网Network of Schools
- 用Java做一个跳一跳辅助
- Ubuntu 16.04安装Wine版的迅雷+QQ(完美方案,终极解决方法)
- 用Redis存储Tomcat集群的Session(转载)
- docker run 报错——WARNING: IPv4 forwarding is disabled. Networking will not work.
- 【sublime text3】破解 最近破解码 /激活成功,但是过一会就提示激活码失效的 Build3143