如何在SymPy表达式中实现数值智能取整？

1. 字典形式解的格式化 #

针对类似{P1: 30.0000000000000, P2: 60.0000000000000}的解，遍历键值对判断数值是否为整数型浮点数，按需转换：

def format_solution(sol_dict):
    result = {}
    for var, val in sol_dict.items():
        if isinstance(val, float) and val.is_integer():
            result[var] = int(val)
        else:
            result[var] = round(val, 2)
    return result

# 示例调用
from sympy import Symbol
P1, P2 = Symbol('P1'), Symbol('P2')
raw_sol = {P1: 30.0000000000000, P2: 60.0000000000000}
formatted_sol = format_solution(raw_sol)
print(formatted_sol)  # 输出 {P1: 30, P2: 60}

2. 表达式系数的格式化 #

针对带科学计数法系数的SymPy表达式，遍历替换每个系数，整数转整型，非整数用科学计数法保留两位小数：

from sympy import preorder_traversal, Float, Integer, Symbol

x = Symbol('x')
raw_expr = 1.40765765765766e-18*x**4 - 5.63063063063063e-15*x**3 + 1.40765765765766e-11*x**2 - 5.63063063063063e-9*x - 1.87687687687688e-7

def format_expr_coefficients(expr):
    for node in preorder_traversal(expr):
        if isinstance(node, Float):
            if node.is_integer():
                expr = expr.subs(node, Integer(int(node)))
            else:
                # 格式化科学计数法为两位小数
                formatted = "{:.2e}".format(node)
                # 去掉末尾多余的0（如1.40e-18 → 1.4e-18）
                formatted = formatted.replace(".00", "").replace(".0", "")
                expr = expr.subs(node, Float(formatted))
    return expr

formatted_expr = format_expr_coefficients(raw_expr)
print(formatted_expr)
# 输出 1.4e-18*x**4 - 5.63e-15*x**3 + 1.4e-11*x**2 - 5.63e-9*x - 1.88e-7

注意事项 #

• 若需要严格匹配示例中1.4e-18这类一位小数的科学计数法，可根据系数的小数部分调整格式化规则（比如对小于1.5e-n的系数保留一位小数），上述代码已覆盖通用场景。
• SymPy的Symbol类型无需额外处理，直接参与替换即可。

内容的提问来源于stack exchange，提问作者علي موسى العامري