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

时间:2016-04-23 12:23:59

标签: python sympy mpmath

我进行了以下导入:

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

似乎是在交流之前设置矩阵时才发生这种情况,这对我来说似乎很奇怪,因为我指出了一个症状复杂类型。我想我可能在某处使用了错误的矩阵或输入类型。

1 个答案:

答案 0 :(得分:2)

问题是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'))