如何用SymPy将表达式中非指数的独立整数替换为1

可以通过SymPy的节点遍历和父节点判断实现精准替换，避免修改指数中的整数。以下是两种简洁的实现方式：

方式一：递归遍历替换 #

from sympy import symbols, Pow, Integer

x = symbols('x')
expr = 2/x**2 + 3/x**3 + 4

def replace_non_exponent_ints(expr):
    def recurse(node):
        # 匹配非1的整数
        if isinstance(node, Integer) and node != 1:
            parent = expr.getparent(node)
            # 若父节点不是幂运算的指数部分，则替换为1
            if parent is None or not (isinstance(parent, Pow) and parent.args[1] is node):
                return Integer(1)
        # 递归处理所有子节点
        return node.func(*[recurse(arg) for arg in node.args])
    return recurse(expr)

# 执行替换并输出结果
result = replace_non_exponent_ints(expr)
print(result)  # 输出：1/x**2 + 1/x**3 + 1

方式二：使用replace方法配合Lambda表达式 #

这是更紧凑的写法，直接利用SymPy的replace方法完成匹配和替换：

from sympy import symbols, Pow, Integer

x = symbols('x')
expr = 2/x**2 + 3/x**3 + 4

result = expr.replace(
    # 匹配非1的整数
    lambda n: isinstance(n, Integer) and n != 1,
    # 仅替换非指数位置的整数
    lambda n: 1 if not (isinstance(n.getparent(expr), Pow) and n.getparent(expr).args[1] is n) else n
)

print(result)  # 输出：1/x**2 + 1/x**3 + 1

核心逻辑说明 #

• 利用getparent方法获取整数节点的父节点，判断该整数是否是Pow（幂运算）的第二个参数（即指数部分）。
• 仅对非指数位置的独立整数执行替换，确保指数中的数字保留不变。

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