如何解决NumPy与SymPy中负整数指数运算报错问题？

问题描述 #

我尝试用SymPy和NumPy测试两个序列是否等价，但运行代码时遇到了负整数指数运算的报错，目前还没找到合适的解决办法，希望能修正代码支持该操作。

原代码：

import numpy
import sympy

n = sympy.symbols('n', integer=True)
a = numpy.arange(1, 100, dtype=float)
expr1 = (-1)**(n-1)*4**(n-1)/3**(n-2)
expr2 = 3*(-4/3)**(n-1)
f = sympy.lambdify(n, expr1, "numpy")
g = sympy.lambdify(n, expr2, "numpy")

# Calculate the arrays using the lambdified functions
array_f = f(a)
array_g = g(a)

# Check if the arrays are equal element-wise
if numpy.array_equal(array_f, array_g):
    print('True')

报错原因 #

报错根源在于expr1中的3**(n-2)运算：当n=1时，指数为-1，虽然a是float类型，但SymPy lambdify生成的代码中，底数3是整数类型，NumPy对整数底数执行负指数运算时会触发类型不兼容问题（整数的负指数结果为浮点数，与原整数底数的运算逻辑冲突）。

修正方案 #

有两种可行的修正方式：

方式1：在SymPy中先化简表达式

先通过SymPy的化简功能将expr1转换为与expr2一致的形式，避免复杂的指数拆分运算：

import numpy
import sympy

n = sympy.symbols('n', integer=True)
a = numpy.arange(1, 100, dtype=float)

# 先化简expr1，确认数学等价性
expr1 = (-1)**(n-1)*4**(n-1)/3**(n-2)
simplified_expr1 = sympy.simplify(expr1)  # 化简后得到3*(-4/3)**(n-1)，与expr2完全一致
expr2 = 3*(-4/3)**(n-1)

f = sympy.lambdify(n, simplified_expr1, "numpy")
g = sympy.lambdify(n, expr2, "numpy")

array_f = f(a)
array_g = g(a)

# 考虑浮点数精度问题，用allclose替代array_equal
if numpy.allclose(array_f, array_g):
    print('True')

方式2：将表达式中的整数底数改为浮点数

把expr1中的整数底数替换为浮点数，确保NumPy能正确处理负指数运算：

import numpy
import sympy

n = sympy.symbols('n', integer=True)
a = numpy.arange(1, 100, dtype=float)

# 将整数底数改为浮点数，避免负指数运算报错
expr1 = (-1.0)**(n-1)*4.0**(n-1)/3.0**(n-2)
expr2 = 3*(-4/3)**(n-1)

f = sympy.lambdify(n, expr1, "numpy")
g = sympy.lambdify(n, expr2, "numpy")

array_f = f(a)
array_g = g(a)

# 浮点数比较建议用allclose，因为存在精度误差
if numpy.allclose(array_f, array_g):
    print('True')

额外说明 #

• 浮点数运算存在精度误差，因此不建议用numpy.array_equal（要求完全相等），改用numpy.allclose（允许指定精度范围内的相等）更合理。
• 两种方式都能解决负指数报错问题，方式1通过化简表达式让代码更简洁，方式2直接修改运算数类型，更直观。

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