如何用SymPy将有理表达式化简为指定形式并调整美化输出？

我最近遇到一个问题：用SymPy处理有理表达式E时，默认的化简方法没法得到我手动推导的目标形式F；同时还需要调整SymPy的pretty print格式，满足两个特定需求。以下是问题详情和解决办法：

问题背景 #

首先定义表达式E的代码：

import sympy as sp
a, b, c, d, n, t, A, B, C = sp.symbols('a, b, c, d, n, t, A, B, C', real = True)
E = n/(c-b) * ( B - (c-b)/(c-a)*A - (b-a)/(c-a)*B ) * (c-t)/(c-b) + n/(c-b) * ( (d-c)/(d-b)*B + (c-b)/(d-b)*C - B ) * (t-b)/(c-b)
print(sp.pretty( E ))
print(sp.pretty( E.simplify() ))

运行后得到的原始表达式和simplify()结果：

⎛ B⋅(-c + d) C⋅(-b + c)⎞ ⎛ A⋅(-b + c) B⋅(-a + b) ⎞
n⋅(-b + t)⋅⎜-B + ────────── + ──────────⎟ n⋅(c - t)⋅⎜- ────────── - ────────── + B⎟
⎝ -b + d -b + d ⎠ ⎝ -a + c -a + c ⎠
───────────────────────────────────────── + ─────────────────────────────────────────
(-b + c)² (-b + c)²

-n⋅((a - c)⋅(b - t)⋅(-B⋅(b - d) + B⋅(c - d) + C⋅(b - c)) + (b - d)⋅(c - t)⋅(A⋅(b - c) + B⋅(a - b) - B⋅(a - c)))
────────────────────────────────────────────────────────────────────────────────────────────────────────────────
(a - c)⋅(b - c)²⋅(b - d)

而手动推导可以得到更简洁的结构F：

F = n/(c-a) * (B - A) * (c-t)/(c-b) + n/(d-b) * (C - B) * (t-b)/(c-b)
print(sp.pretty( F ))
print((F-E).simplify())

对应的输出验证了F和E等价：

n⋅(-A + B)⋅(c - t) n⋅(-B + C)⋅(-b + t)
──────────────────── + ───────────────────
(-a + c)⋅(-b + c) (-b + c)⋅(-b + d)

0

试过factor()、collect()和apart()等方法，都没能得到和F一致的结构，同时还需要调整pretty print的两个格式问题：

1. 保留分子分母中变量的原始顺序（比如显示B - A而非-A + B）
2. 将复合分数显示为简单分数的乘积（比如a/b·c/d而非ac/bd）

解决办法 #

一、将E化简为F的结构 #

SymPy默认的simplify()会倾向于合并成单个大分式，我们可以通过分步处理引导它生成目标结构：

方法1：拆分表达式分别化简

把E拆成两个子项E1和E2，单独化简后再组合：

# 拆分E为两个子项
E1 = n/(c-b) * ( B - (c-b)/(c-a)*A - (b-a)/(c-a)*B ) * (c-t)/(c-b)
E2 = n/(c-b) * ( (d-c)/(d-b)*B + (c-b)/(d-b)*C - B ) * (t-b)/(c-b)

# 分别化简子项并整理成目标形式
E1_simplified = sp.simplify(E1)
# 手动调整符号顺序和分式结构
E1_final = n*(B - A)*(c - t)/((c - a)*(c - b))

E2_simplified = sp.simplify(E2)
E2_final = n*(C - B)*(t - b)/((d - b)*(c - b))

# 合并得到F的结构
result = E1_final + E2_final
print(sp.pretty(result))

方法2：展开后按指定项分组

先展开E，再按(c-t)和(t-b)分组收集，最后整理符号：

E_expanded = sp.expand(E)

# 收集(c-t)相关的项并化简
term_ct = sp.collect(E_expanded, (c - t)).coeff(c - t) * (c - t)
term_ct_simplified = sp.simplify(term_ct).subs(-A + B, B - A).subs(-a + c, c - a).subs(-b + c, c - b)

# 收集(t-b)相关的项并化简
term_tb = sp.collect(E_expanded, (t - b)).coeff(t - b) * (t - b)
term_tb_simplified = sp.simplify(term_tb).subs(-B + C, C - B).subs(-b + d, d - b).subs(-b + c, c - b)

# 合并结果
result = term_ct_simplified + term_tb_simplified
print(sp.pretty(result))

二、调整SymPy Pretty Print格式 #

1. 保留变量原始顺序（如B - A而非-A + B）

SymPy默认按字母顺序排列项，我们可以通过手动替换或自定义打印机实现：

手动替换法（简单高效）

针对特定的符号组合直接替换：

# 假设已经得到化简后的表达式
expr = n*(-A + B)*(c - t)/((-a + c)*(-b + c)) + n*(-B + C)*(-b + t)/((-b + c)*(-b + d))
# 替换成我们想要的符号顺序
expr_rewritten = expr.subs(-A + B, B - A)\
                    .subs(-B + C, C - B)\
                    .subs(-a + c, c - a)\
                    .subs(-b + d, d - b)\
                    .subs(-b + c, c - b)\
                    .subs(-b + t, t - b)
print(sp.pretty(expr_rewritten))

自定义打印机法（通用场景）

继承PrettyPrinter并重写加法项的排序逻辑：

from sympy.printing.pretty import PrettyPrinter

class CustomOrderPrinter(PrettyPrinter):
    def _print_Add(self, expr):
        # 自定义项的排序优先级，优先显示我们想要的组合
        def sort_key(term):
            if term.has(B - A):
                return 0
            elif term.has(C - B):
                return 1
            # 其他项按默认顺序
            return super()._print_Add(expr)
        
        # 按自定义规则排序后再打印
        sorted_terms = sorted(expr.args, key=sort_key)
        return super()._print_Add(sp.Add(*sorted_terms))

# 使用自定义打印机
printer = CustomOrderPrinter()
printer.pprint(expr_rewritten)

2. 将复合分数显示为分数乘积

SymPy默认会合并分式，最直接的方法是按目标结构定义表达式，而非先合并再拆分：

# 直接写成多个分数相乘的形式
F = n/(c-a) * (B - A) * (c-t)/(c-b) + n/(d-b) * (C - B) * (t-b)/(c-b)
print(sp.pretty(F))

这种方式下SymPy会自动显示为分数的乘积形式，无需额外处理。如果需要对已有的合并分式进行拆分，可以手动将分子的乘积拆分为多个分式相乘的结构。

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