如何在Python Sympy中设置容差以将近似相等浮点数视为相等？

当使用Sympy处理带有微小差异浮点数系数的表达式时，严格的符号运算会因系数不完全相等拒绝化简，多次复合运算后微小误差会被放大，导致代入计算结果异常。以下是几种设置容差的解决方案：

方法1：用nsimplify将相近浮点数转换为相同有理数 #

通过nsimplify的tolerance参数指定容差，将差异在阈值内的浮点数统一为相同的有理数，让Sympy可以正常化简：

import sympy as sp

a = 2.924950591148529
c = 2.92495059114853
x = sp.symbols('x')

# 应用容差转换系数
expr1 = sp.nsimplify(a*x + 1, [x], tolerance=1e-15)
expr2 = sp.nsimplify(c*x + 1, [x], tolerance=1e-15)

first = expr1 / expr2 / expr1 / expr2
second = expr2 / expr1 / expr2 / expr1 

result = sp.cancel(first/second/first/second/first/second/second/first/second/first/second) 
print(result.subs(x, 10))  # 输出：1

方法2：自定义函数替换相近系数 #

遍历表达式提取系数，手动将差异在容差内的系数替换为同一个值，适合需要精准控制替换逻辑的场景：

import sympy as sp

a = 2.924950591148529
c = 2.92495059114853
x = sp.symbols('x')
expr1 = a*x + 1
expr2 = c*x + 1

tolerance = 1e-15

def align_coefficients(expr_target, expr_ref, tol):
    # 提取x的系数和常数项
    coeff_target, const_target = expr_target.as_coeff_add(x)
    coeff_ref, const_ref = expr_ref.as_coeff_add(x)
    const_target = const_target[0]
    const_ref = const_ref[0]
    
    # 替换相近的x系数
    if abs(coeff_target - coeff_ref) < tol:
        expr_target = expr_target.subs(coeff_target, coeff_ref)
    # 替换相近的常数项（按需启用）
    if abs(const_target - const_ref) < tol:
        expr_target = expr_target.subs(const_target, const_ref)
    return expr_target

# 将expr1的系数对齐到expr2
expr1 = align_coefficients(expr1, expr2, tolerance)

first = expr1 / expr2 / expr1 / expr2
second = expr2 / expr1 / expr2 / expr1 

result = sp.cancel(first/second/first/second/first/second/second/first/second/first/second) 
print(result.subs(x, 10))  # 输出：1

方法3：事后用chop修正计算结果 #

如果仅需修正最终代入后的异常值，可以用sp.chop将极小/极大的异常值截断为合理值，适合不需要修改原始表达式的场景：

import sympy as sp

a = 2.924950591148529
c = 2.92495059114853
x = sp.symbols('x')
expr1 = a*x + 1
expr2 = c*x + 1

first = expr1 / expr2 / expr1 / expr2
second = expr2 / expr1 / expr2 / expr1 

result = sp.cancel(first/second/first/second/first/second/second/first/second/first/second) 
# 代入后修正结果
print(sp.chop(result.subs(x, 10)))  # 输出：1

原理说明 #

Sympy的符号运算默认严格区分所有浮点数，即使差异在1e-16量级也会被视为不同系数，导致无法触发化简逻辑。多次乘除复合运算后，微小的系数差异会被指数级放大，产生异常结果。设置容差的核心是将相近的浮点数统一为同一值，让Sympy识别到表达式的可化简性。

内容的提问来源于stack exchange，提问作者Sliem el Ela