Python 中的枚举是一种用户定义的数据类型,由一组命名值组成。的有限集合值是使用枚举定义的,并且可以在 Python 中使用它们的名称而不是整数值来访问这些值。枚举使代码更具可读性和可维护性,并且还增强了类型安全性。在本文中,我们将了解如何在 Python 中通过字符串值查找枚举。
要通过字符串值查找枚举,我们需要按照以下步骤进行:
-
在代码中导入枚举模块
-
定义具有所需值集的枚举
-
创建一个函数,将枚举字符串作为输入并返回相应的枚举值。
语法
from enum import Enum
class ClassName(Enum):
Key_1= Value_1
Key_2= Value_2
Key_3= Value_3
登录后复制
在Python中使用枚举需要导入枚举,然后创建一个类,该类将以枚举值作为输入,并且还包含每个枚举值的键值对。
使用枚举提高代码可读性
除了通过名称查找值之外,Enum 还可以提高代码的可读性。
示例
下面的代码由一个名为 process_color() 的函数组成,该函数接受一个整数作为输入并返回指示正在处理的颜色的消息。下面的代码不被认为是一个好的可读代码,因为我们必须记住每个整数值代表哪种颜色。我们可以使用枚举来提高代码的可读性。
def process_color1(color):
if color == 1:
print("Processing red color")
elif color == 2:
print("Processing green color")
elif color == 3:
print("Processing blue color")
else:
raise ValueError(f"{color} is not a valid color")
登录后复制
使用枚举,我们可以定义一个名为color的枚举,其值与上述代码中的值相同。然后通过创建一个名为process_color的简单函数,该函数以整数作为输入,并使用枚举将整数转换为命名的颜色值。如果输入不是有效的颜色,则引发一个带有详细错误消息的ValueError。这使得代码更易读,因为我们现在不必记住每个颜色的整数值。
class Color(Enum):
RED = 1
GREEN = 2
BLUE = 3
def process_color(color):
try:
color = Color(color)
except ValueError:
raise ValueError(f"{color} is not a valid color")
print(f"Processing {color.name.lower()} color")
登录后复制
结论
在本文中,我们了解了如何使用命名字符串查找枚举值。枚举可以使代码更易读并提高代码的可维护性。在涉及有限命名值集的任何项目中都应考虑使用枚举。
以上就是Python程序通过字符串值查找枚举的详细内容,更多请关注每日运维网(www.mryunwei.com)其它相关文章!