压缩一个Python方法,该方法根据传递的运算符进行不同的比较

使用 operator 模块,为所有标准运算符提供命名函数。然后,您可以使用字典将运算符字符串映射到相应的函数。

from operator import lt, gt, eq

ops = {'<': lt, '>': gt, '=': eq}

def comparsion(val1: int, val2: int, operator: str):
    return ops[operator](val1, val2)

用字典帮你处理事情怎么样?

def comparsion(val1: int, val2: int, operator: str):
    operations = {
        ">": val1 > val2,
        "=": val1 == val2,
        "<": val1 < val2
    }
    return operations[operator]

print(comparsion(1, 2, ">"))
print(comparsion(1, 2, "="))
print(comparsion(1, 2, "<"))

如果事情变得更复杂,字典中的值可能会被提升为lambdas。

如果只有三个操作员,我就不会尝试更复杂的操作。也许如果还有更多,你可以尝试一个字典,其中键是运算符,值是比较函数。

这工作得很好:

def comp(val1,val2,op):
    return eval(f"{val1}{op}{val2}")