在排序数组时,我们可以使用自定义的比较函数来按枚举的名称而不是值进行排序。下面是一个示例代码:
# 导入枚举模块
from enum import Enum
# 定义枚举类
class Color(Enum):
RED = 1
GREEN = 2
BLUE = 3
# 待排序的数组
colors = [Color.GREEN, Color.RED, Color.BLUE]
# 自定义比较函数,按枚举的名称进行排序
def compare_by_name(color1, color2):
return color1.name > color2.name
# 使用自定义比较函数进行排序
sorted_colors = sorted(colors, key=lambda x: x, cmp=compare_by_name)
# 输出排序结果
for color in sorted_colors:
print(color.name)
在上述代码中,我们首先定义了一个枚举类Color
,其中包含了三个枚举值RED
、GREEN
和BLUE
。然后我们定义了一个待排序的数组colors
,其中包含了这三个枚举值。
接下来,我们定义了一个自定义的比较函数compare_by_name
,该函数会比较两个枚举值的名称,然后根据名称的大小关系返回比较结果。
最后,我们使用sorted
函数来对colors
数组进行排序,其中通过key
参数指定了按枚举值本身进行排序,而通过cmp
参数指定了使用自定义的比较函数进行排序。排序结果保存在sorted_colors
中。
最后,我们遍历输出了排序后的结果,可以看到数组中的枚举值按名称的字典序进行了排序。输出结果为:
BLUE
GREEN
RED