Python程序通过字符串值查找枚举-Python教程-PHP中文网

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

WBOY

发布： 2023-09-21 21:25:10

转载

1383人浏览过

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

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")

登录后复制