Python 通过字符串值查找枚举

在Python中，枚举是一种用户自定义的数据类型，由一组命名值组成。可以使用枚举来定义一组有限的值，而这些值可以通过它们的名字而不是整数值在Python中访问。枚举使代码更可读，更易维护，还可以强制类型安全。在本文中，我们将看到如何在Python中通过字符串值查找枚举。

要通过字符串值查找枚举，需要遵循以下步骤：

在代码中导入枚举模块
使用所需值定义枚举
创建一个函数，该函数接受枚举字符串作为输入并返回相应的枚举值。

语法

from enum import Enum

class ClassName(Enum):
   Key_1= Value_1
   Key_2= Value_2
   Key_3= Value_3

在Python中使用枚举类型，我们需要导入枚举并创建一个类，该类将以枚举值作为输入，并且还包含每个枚举值的键值对。

使用枚举提高代码的可读性

除了按名称查找值之外，枚举还可以提高代码的可读性。

示例

下面的代码包含一个名为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")