python 用枚举来作为函数的参数

在 Python 中，可以使用枚举（Enum）来作为函数的参数。枚举是一种定义常量的方式，它将一组常量封装在一起，可以方便地进行比较和判断。

下面是一个简单的例子，演示如何在函数中使用枚举：

from enum import Enum

class Colors(Enum):
    RED = 1
    GREEN = 2
    BLUE = 3

def print_color(color):
    if color == Colors.RED:
        print("Red")
    elif color == Colors.GREEN:
        print("Green")
    elif color == Colors.BLUE:
        print("Blue")

print_color(Colors.RED)

在这个例子中，我们定义了一个枚举 `Colors`，其中包含了三种颜色：红色、绿色和蓝色。然后定义了一个函数 `print_color`，它接受一个参数 `color`，该参数的类型是 `Colors` 枚举类型。在函数内部，我们通过比较枚举类型的值来判断颜色，并打印出相应的字符串。

最后，我们调用函数 `print_color`，并传入 `Colors.RED` 作为参数，函数会打印出 "Red"。