调用 qr_solve 时无法从复数创建 mpf

Question

我进行了以下导入：

from sympy.matrices import Matrix as sy_matrix
import sympy.mpmath as sy_mp

像这样创建我的矩阵：

sysMat = sy_matrix([[0.0]*sz1]*sz2)       
resVec = sy_matrix([[0.0]]*sz2)

。然后用 python 复数（例如 1.0+1.0j）填充，然后调用：

coeffVec = sy_mp.qr_solve(sysMat, resVec)

但是我得到以下异常：

  File "..\RatSMat.py", line 69, in solve
    self.coeffVec = sy_mp.qr_solve(self.sysMat, self.resVec)
  File "C:\Python27\lib\site-packages\sympy\mpmath\matrices\linalg.py", line 406, in qr_solve
    A, b = ctx.matrix(A, **kwargs).copy(), ctx.matrix(b, **kwargs).copy()
  File "C:\Python27\lib\site-packages\sympy\mpmath\matrices\matrices.py", line 328, in __init__
    A = self.ctx.matrix(args[0].tolist())
  File "C:\Python27\lib\site-packages\sympy\mpmath\matrices\matrices.py", line 301, in __init__
    self[i, j] = convert(a)
  File "C:\Python27\lib\site-packages\sympy\mpmath\ctx_mp_python.py", line 662, in convert
    return ctx._convert_fallback(x, strings)
  File "C:\Python27\lib\site-packages\sympy\mpmath\ctx_mp.py", line 614, in _convert_fallback
    raise TypeError("cannot create mpf from " + repr(x))
  TypeError: cannot create mpf from -2.27448396079674e-13*I

它似乎发生在 sympy 在输入之前设置矩阵时，这对我来说似乎很奇怪，因为 I 表示一个 sympy 复杂类型。我想我可能在某处使用了错误的矩阵或输入类型。

Answer 1

问题是 mpmath 不知道如何将 Sympy 对象 I（虚数单位）转换为其表示复数的对象。不过，它可以与 Python 的 j 一起使用，如您所见 in the source。

需要注意的是，mpmath是一个高精度浮点计算库，它独立于符号计算库Sympy。无需定义 Sympy 对象，然后让 mpmath 使用它们。直接用mpmath就可以了。

import mpmath
mpmath.mp.dps = 50    # 50 digit arithmetic, for demonstration
A = mpmath.matrix([[2+3j, 1], [1, 1]])
b = mpmath.matrix([[1], [1]])
mpmath.qr_solve(A, b)

输出：

(matrix(
[[mpc(real='3.436455137311543569947658013827436034380299778494123e-55', imag='-1.2867118438861911327915233933248665673255859691704302e-55')],
 [mpc(real='1.0', imag='-1.1953121943036561823547019562093175737721941131313411e-54')]]), mpf('1.7055229691470039210052256620915903242996742650953353e-54'))

调用 qr_solve 时无法从复数创建 mpf

Cannot create mpf from a complex number when calling qr_solve

python

sympy

mpmath