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("处理红色")
   elif color == 2:
      print("处理绿色")
   elif color == 3:
      print("处理蓝色")
   else:
      raise ValueError(f"{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}不是有效的颜色")
   print(f"处理{color.name.lower()}颜色")