Python如何基于自定义优先级规则比较字符串变量大小？

核心规则对齐需求：优先级顺序为 B > A > D > C，优先级越高的字符串，在比较时判定为“更大”，保证 D > B 这类判断返回False，最终进入else分支。

方案1：轻量局部实现——优先级映射表 #

适合临时使用的简单场景，不需要复杂封装：

• 提前建立字符串到优先级数值的映射，数值越高代表优先级越高
• 比较时直接取两个字符串对应的优先级数值做对比，逻辑清晰

# 定义优先级映射
PRIORITY = {
    'B': 4,
    'A': 3,
    'D': 2,
    'C': 1
}

# 示例场景
var1 = 'D'
var2 = 'B'
if PRIORITY[var1] > PRIORITY[var2]:
    pass # 不会进入该分支
else:
    pass # 命中该分支，符合预期

优缺点：写法简单无额外依赖，但每次比较需要手动取映射值，无法直接对原始字符串使用>、<运算符。

方案2：完全适配需求——自定义可比较字符串类 #

如果要实现和数值类型完全一致的直接比较体验，可以自定义继承自str的类，通过重写富比较方法实现自定义比较逻辑：

from functools import total_ordering

PRIORITY = {
    'B': 4,
    'A': 3,
    'D': 2,
    'C': 1
}

@total_ordering
class PriorityStr(str):
    def __new__(cls, value):
        if value not in PRIORITY:
            raise ValueError(f"不支持的取值{value}，可选值为{list(PRIORITY.keys())}")
        return super().__new__(cls, value)
    
    def __lt__(self, other):
        if not isinstance(other, PriorityStr):
            return NotImplemented
        return PRIORITY[self] < PRIORITY[other]
    
    def __eq__(self, other):
        if not isinstance(other, PriorityStr):
            return NotImplemented
        return PRIORITY[self] == PRIORITY[other]

使用说明：

• 类实例完全兼容原生字符串的所有方法，可直接当普通字符串使用
• 借助@total_ordering装饰器，仅实现__lt__和__eq__两个方法，即可自动生成>、>=、<=等全量比较运算符的逻辑，无需重复编码
• 内置取值校验，可拦截未定义优先级的字符串参与比较，减少逻辑错误

# 测试效果
var1 = PriorityStr('D')
var2 = PriorityStr('B')
if var1 > var2:
    pass # 不会进入
else:
    pass # 命中，符合预期

方案3：无侵入全局适配——猴子补丁修改原生str比较逻辑 #

如果项目中已有大量普通字符串变量，不想逐个替换为自定义类，可以在可控范围内修改原生str的比较行为：

PRIORITY = {
    'B': 4,
    'A': 3,
    'D': 2,
    'C': 1
}

# 保存原生比较方法，避免完全覆盖原有逻辑
original_str_lt = str.__lt__
original_str_eq = str.__eq__

def custom_lt(self, other):
    # 仅当两个值都在自定义优先级范围内时，走自定义比较逻辑
    if self in PRIORITY and other in PRIORITY:
        return PRIORITY[self] < PRIORITY[other]
    # 其他场景保留原生字符串的字典序比较逻辑
    return original_str_lt(self, other)

def custom_eq(self, other):
    if self in PRIORITY and other in PRIORITY:
        return PRIORITY[self] == PRIORITY[other]
    return original_str_eq(self, other)

# 替换全局str的比较方法
str.__lt__ = custom_lt
str.__eq__ = custom_eq

注意事项：

• 这种方式会影响全局所有字符串的比较逻辑，仅适合在项目完全可控、无其他依赖原生字符串字典序比较的场景下使用
• 未在优先级映射内的字符串，依然保持原生的字典序比较逻辑，不会破坏原有功能
• 修改后可直接使用普通字符串做比较，和方案2的效果完全一致，无需逐个替换变量类型。

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