如何强制SymPy将表达式统一转换为仅含负指数或分数形式？

在SymPy中处理包含除法和乘法的表达式时，展开后的打印输出会出现不一致的格式：当项的系数为1时，SymPy会用负指数形式（如x**(-2)）显示；系数非1时则用分数形式（如3/x**2）。例如：

from sympy import symbols, expand
x = symbols('x')

print(expand((3+x)/x/x))  # 输出: 1/x + 3/x**2
print(expand((1+x)/x/x))  # 输出: 1/x + x**(-2)

一、转换为仅含负指数的形式 #

要让所有项统一以负指数形式呈现，可通过以下两种方式实现：

方式1：自定义打印函数 #

直接解析表达式的系数与指数，拼接成负指数格式的输出字符串：

from sympy import symbols, expand

def print_with_negative_exponents(expr, var):
    terms = expr.as_ordered_terms()
    output_terms = []
    for term in terms:
        coeff, exp = term.as_coeff_exponent(var)
        if coeff == 1:
            output_terms.append(f"x**({exp})")
        elif coeff == -1:
            output_terms.append(f"-x**({exp})")
        else:
            output_terms.append(f"{coeff}*x**({exp})")
    # 处理负号格式
    return " + ".join(output_terms).replace("+ -", "- ")

# 测试
x = symbols('x')
e1 = expand((3+x)/x/x)
e2 = expand((1+x)/x/x)
print(print_with_negative_exponents(e1, x))  # 输出: x**(-1) + 3*x**(-2)
print(print_with_negative_exponents(e2, x))  # 输出: x**(-1) + x**(-2)

方式2：修改表达式内部结构 #

如果需要让表达式本身以负指数形式存储（而非仅修改打印格式），可以用replace方法批量转换分式项：

from sympy import symbols, expand, Pow

def convert_to_negative_exponents(expr):
    # 将所有 k/x^n 形式的项转换为 k*x**(-n)
    return expr.replace(
        lambda t: isinstance(t, Pow) and t.exp.is_Positive,
        lambda t: Pow(t.base, -t.exp)
    ).replace(
        lambda t: isinstance(t, Mul) and any(isinstance(arg, Pow) and arg.exp.is_Positive for arg in t.args),
        lambda t: Mul(*[arg if not (isinstance(arg, Pow) and arg.exp.is_Positive) else Pow(arg.base, -arg.exp) for arg in t.args])
    )

# 测试
x = symbols('x')
e = expand((3+x)/x/x)
converted_expr = convert_to_negative_exponents(e)
print(converted_expr)  # 输出: x**(-1) + 3*x**(-2)

二、转换为仅含分数的形式 #

要让所有项统一以分数形式呈现，只需将负指数项替换为分式即可：

from sympy import symbols, expand, Pow

def convert_to_fraction_form(expr):
    # 将所有 x**(-n) 形式的项转换为 1/x**n
    return expr.replace(
        lambda t: isinstance(t, Pow) and t.exp.is_Negative,
        lambda t: 1/(t.base**(-t.exp))
    )

# 测试
x = symbols('x')
e = expand((1+x)/x/x)
converted_expr = convert_to_fraction_form(e)
print(converted_expr)  # 输出: 1/x + 1/x**2

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