如何将数值比较结果（>、=、<）作为字典键替代冗余分支代码？

Python里没法直接把>/=/<这类运算符作为字典键（它们不是可哈希的对象，字典键要求必须是不可变、可哈希的类型），但可以通过转换比较结果为可哈希的状态标识，实现用字典映射替代冗余的if-elif-else分支。

最简实现：用数值标识状态 #

利用(a > b) - (a < b)可以直接得到三个对应状态的数值：

• 当a > b时，结果为1
• 当a == b时，结果为0
• 当a < b时，结果为-1

用这三个数值作为字典键，就能完成映射：

a = 7
b = 4

compare_map = {
    1: f"{a} > {b}",
    0: f"{a} = {b}",
    -1: f"{a} < {b}"
}

# 计算状态标识
status_code = (a > b) - (a < b)
print(compare_map[status_code])  # 输出：7 > 4

更具可读性的实现：用枚举类定义状态 #

如果觉得数值标识不够直观，可以用枚举类明确定义比较状态，代码可读性更强：

from enum import Enum

class CompareResult(Enum):
    GREATER = "greater"
    EQUAL = "equal"
    LESS = "less"

def get_compare_result(a, b):
    if a > b:
        return CompareResult.GREATER
    elif a == b:
        return CompareResult.EQUAL
    else:
        return CompareResult.LESS

a = 5
b = 5
compare_map = {
    CompareResult.GREATER: f"{a} > {b}",
    CompareResult.EQUAL: f"{a} = {b}",
    CompareResult.LESS: f"{a} < {b}"
}

print(compare_map[get_compare_result(a, b)])  # 输出：5 = 5

对比原分支代码的优势 #

这种字典映射的方式，把判断逻辑和业务逻辑（这里是格式化输出）分离开：

• 后续如果要修改输出格式，只需要修改字典里的字符串模板
• 如果要扩展更多比较相关的操作，直接在字典里新增键值对即可，不需要嵌套更多的elif分支

内容的提问来源于stack exchange，提问作者Dyson