data_factory/utils/compare_values_utils.py

39 lines
1.1 KiB
Python

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time : 2025/3/20 19:29
# @Author : AngesZhu
# @File : compare_values_utils.py
# @Desc : 运算符对比函数
def compare_values(a, b, op_name):
"""
根据传入的运算符名称比较两个值。
:param a: 第一个值
:param b: 第二个值
:param op_name: 运算符名称(如 "gt", "lt", "eq" 等)
:return: 比较结果
"""
operations = {
"gt": lambda x, y: x > y,
"lt": lambda x, y: x < y,
"eq": lambda x, y: x == y,
"ge": lambda x, y: x >= y,
"le": lambda x, y: x <= y,
"ne": lambda x, y: x != y,
}
if op_name not in operations:
raise ValueError(f"Unsupported operation: {op_name}")
return operations[op_name](a, b)
if __name__ == "__main__":
# 定义两个值
value1 = 10
value2 = 5
# 使用字符串指定运算符
print("10 > 5:", compare_values(value1, value2, "gt")) # 大于
print("10 < 5:", compare_values(value1, value2, "lt")) # 小于
print("10 == 5:", compare_values(value1, value2, "eq")) # 等于