详解Python中的枚举类型
你好，我是 征哥，今天分享一下 Python 中的枚举类型，为什么需要枚举类型，及如何使用。
什么是枚举类型
枚举(Enum)是一种数据类型，是绑定到唯一值的符号表示。您可以使用它来创建用于变量和属性的常量集。它们类似于全局变量，但是，它们提供了更有用的功能，例如分组和类型安全。Python 在 3.4 版本中添加了标准库 enum。
为什么要使用枚举
使用枚举有以下好处：
代码更容易阅读，更容易维护。
减少由转换或错误输入引起的 bug。
使将来修改值变得更容易。
如何使用枚举
以我们最熟悉的性别为例，先创建一个枚举类型：
>>> from enum import Enum >>> class Gender(Enum): ... 男 = 1 ... 女 = 0 ... 未知 = 2 ... >>> Gender <enum 'Gender'>
访问枚举：
>>> for g in Gender: ... print(g.name,g.value) ... 男 1 女 0 未知 2 >>> print(Gender.男.name) 男 >>> print(Gender.男.value) 1 >>> isinstance(Gender.男,Gender) True >>> Gender['男'] <Gender.男: 1> >>> Gender['男'].value 1 >>> Gender['男'].name '男' >>> Gender(2).name '未知'
接下来，让我们先看看不使用枚举的版本，然后再看看使用枚举的版本，这样就知道枚举的好处了。
以汽车销售为例，我们需要根据不同国家的税率来计算汽车的出售价格。不使用枚举的版本如下：
from dataclasses import dataclass
@dataclass
class Car:
model: str
price: float
registraion_state: str
def total_cost(self) -> float:
match self.registraion_state:
case "OR":
return self.price + (self.price * 0.05)
case "WA":
return self.price + (self.price * 0.10)
case "CA":
return self.price + (self.price * 0.08)
case _:
raise TypeError("Invalid registraion_state value")
car1 = Car(model="RAV4", price=30000, registraion_state="OR")
car2 = Car(model="RAV4", price=30000, registraion_state="WA")
car3 = Car(model="RAV4", price=30000, registraion_state="CA")
print(car1.total_cost())
print(car2.total_cost())
print(car3.total_cost())
代码共 26 行，用 3.10 版本运行结果如下：
31500.0
33000.0
32400.0
接下来，我们来看一看使用枚举版本的代码：
from dataclasses import dataclass from enum import Enum class StateTax(Enum): OR = 0.05 WA = 0.10 CA = 0.08 @dataclass class Car: model: str price: float tax: StateTax def total_cost(self) -> float: return self.price + (self.price * self.tax.value) def get_tax(self): return self.tax.value car1 = Car(model="RAV4", price=30000, tax=StateTax.OR) car2 = Car(model="RAV4", price=30000, tax=StateTax.WA) car3 = Car(model="RAV4", price=30000, tax=StateTax.CA) print(car1.total_cost()) print(car2.total_cost()) print(car3.total_cost())
代码共 28 行，运行结果不变，但是代码更优雅，更容易阅读了。而且后续增加出口的国家地区，也非常简单，只需要增加一个枚举类型，创建一个新的 Car 对象，需求修改 Car 类的任何代码：
class StateTax(Enum): OR = 0.05 WA = 0.10 CA = 0.08 CN = 0.03 car4 = Car(model="RAV4", price=30000, tax=StateTax.CN)
从字典创建枚举
from enum import Enum
tax = {
'OR': 0.05,
'WA': 0.10,
'CA': 0.08,
'CN': 0.03
}
StateTax = Enum('tax', tax)
访问枚举的三种方式：
print(StateTax['CN']) print(StateTax.CN) print(StateTax(0.03)) #通过 value 来访问
最后的话
枚举 (Enum) 是绑定到唯一值的符号。我们可以通过定义一个继承自 Enum 的类来创建枚举，使用 enum[member_name] 通过名称访问 Enum 成员，也可以通过并通过 enum(member_value) 访问，还可以通过 enum.member_name 来访问。Enum 可以帮助我们使代码更具可读性，更具可维护性，推荐大家都用一用。
到此这篇关于详解Python中的枚举类型的文章就介绍到这了,更多相关Python枚举类型内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
这篇文章主要为大家介绍了Python编程中pydantic会触发及发生访问错误的处理，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步2021-09-09
这篇文章主要介绍了python opencv将多个图放在一个窗口,本文通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2022-02-02
今天小编就为大家分享一篇对Python捕获控制台输出流的方法详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-01-01
本文将以基于Python的车牌识别系统实现为方向,介绍车牌识别技术的基本原理、常用算法和方法,并详细讲解如何利用Python语言实现一个完整的车牌识别系统,需要的朋友可以参考下2023-10-10
python3的url编码和解码,自定义gbk、utf-8的例子
今天小编就为大家分享一篇python3的url编码和解码,自定义gbk、utf-8的例子，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-08-08
今天小编就为大家分享一篇pytorch使用指定GPU训练的实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-08-08
python实现TCPserver是一件简单的事情,只要通过socket这个模块就可以实现,本文就来介绍一下python实现TCPserver的使用示例,感兴趣的可以了解一下2023-10-10
做数据分析的时候,用到了对Excel中的数据进行显示处理,能更直观的了解数据,所以下面这篇文章主要给大家介绍了关于利用python如何绘制表格不同颜色excel的相关资料,需要的朋友可以参考下2021-11-11
AES算法是高级加密标准,它是一种对称加密算法,AES只有一个密钥,这个密钥既用来加密,也用于解密,这篇文章主要给大家介绍了关于python实现aes加密及pycryptodome库使用的相关资料,需要的朋友可以参考下2023-10-10
这篇文章主要介绍了django 外键创建注意事项说明，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-05-05
最新评论