如何将Sympy虚数单位I转为NumPy的j并化简复数符号表达式

问题场景 #

• 现有两个存储复数的NumPy数组a、b，NumPy中虚部单位采用j标识，数组生成代码如下：

for i in range(0, 8):
    cg1 = np.array(eigVal1[i])
    det_eval = eval(det)
    AA = det_eval[0:2,0:2]
    bb = det_eval[0:2,2] * (-1)
    roots = np.append(roots, solve(AA, bb))
    a = np.append(a, roots[i])
    b = np.append(b, roots[i+1])

• 代码运行后得到的数组输出为：

a = array([-9.03839731e-04+0.00091541j,  3.02435614e-07-0.00043776j,
   -9.03839731e-04-0.00091541j,  3.02435614e-07+0.00043776j,
    9.03812649e-04+0.00092323j,  4.17553402e-07+0.00043764j,
    9.03812649e-04-0.00092323j,  4.17553402e-07-0.00043764j])
b = array([ 3.02435614e-07-0.00043776j, -9.03839731e-04-0.00091541j,
    3.02435614e-07+0.00043776j,  9.03812649e-04+0.00092323j,
    4.17553402e-07+0.00043764j,  9.03812649e-04-0.00092323j,
    4.17553402e-07-0.00043764j, -5.53769989e-05-0.00243369j])

• 已构建包含符号变量y的长符号表达式，形式如下：

u_n = A0*y**(1322.5696672125 + 1317.38942049453*I) + A1*y**(1322.5696672125 - 1317.38942049453*I) + A2*y**(-1322.5696672125 + 1317.38942049453*I) + A3*y**(-1322.5696672125 - 1317.38942049453*I) + ..

• 执行系数替换时，所有复数的虚部单位自动变为SymPy默认的I，即便调用simplify(u_n)也无法进一步化简，替换代码如下：

for i in range(0, 8):
    u_n = u_n.subs(A[i], (a[i] * C[i]))

• 替换后得到的表达式形式为：

u_n = C0*y**(1322.5696672125 + 1317.38942049453*I)*(-0.000903839731101097 + 0.000915407724097998*I) + C1*y**(1322.5696672125 - 1317.38942049453*I)*(3.02435613673241e-7 - 0.000437760318205723*I) +..

核心问题：NumPy环境下复数运算可直接合并化简，例如(2+3j)*(5+6j)可直接计算得到(-8+27j)；但SymPy符号表达式中，形如y**(2+3j)*(5+6j)的项会转换为y**(2+3*I)*(5+6*I)的形式，无法自动合并化简，需要实现即使y为符号变量，也能得到复数项合并后的形式（例如y**(-8+27j)）。

解决步骤 #

• 第一步：统一复数类型，消除NumPy与SymPy的类型兼容问题
  直接传入NumPy的complex类型数值会导致SymPy无法识别类型、无法触发自动化简，需要先将NumPy复数显式转换为SymPy原生复数类型，转换示例：

  import numpy as np
  import sympy as sp
  
  def np_c2sp(np_complex):
      # 将numpy复数拆分为实部、虚部，转为SymPy Float类型后组合为SymPy复数
      return sp.Float(np_complex.real) + sp.Float(np_complex.imag)*sp.I
  
  # 替换系数时使用转换后的值
  for i in range(8):
      coeff = np_c2sp(a[i]) * C[i]
      u_n = u_n.subs(A[i], coeff)
  

• 第二步：声明符号属性，消除指数化简的多值歧义
  SymPy默认将符号视为复数域变量，指数、对数运算会保留多值分支无法化简，需要提前将y声明为正实数：

  y = sp.symbols('y', positive=True, real=True)
  

• 第三步：显式调用指数合并逻辑完成化简
  系数与同底指数项合并的本质是利用恒等式k * y^z = y^(z + sp.log(k)/sp.log(y))，直接调用powsimp即可自动完成同底幂的合并：

  # 先展开表达式，拆分所有乘积项
  u_n_expand = sp.expand(u_n)
  # 强制合并同底指数项
  u_n_simplified = sp.powsimp(u_n_expand, force=True)
  

• 可选：如果需要用j作为虚部单位替代SymPy默认的I，最后执行一次符号替换即可：

  j = sp.symbols('j')
  u_n_final = u_n_simplified.subs(sp.I, j)
  

注意：如果表达式中存在需要保留分支的共轭复数、多值函数项，不要开启powsimp的force=True参数，手动逐项提取公因子合并即可，避免出现逻辑错误。

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