Python程序通过字符串值查找枚举

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程序通过字符串值查找枚举的详细内容。更多信息请关注PHP中文网其他相关文章！