我有以下代码来提取对称二阶张量的平方根。
from sympy import symbols, Matrix, mpmath
import numpy as np
F11, F12, F13, F21, F22, F23, F31, F32, F33 = symbols('F11, F12, F13, F21, F22, F23, F31, F32, F33', real=True)
F = np.array([[F11, F12, F13], [F21, F22, F23], [F31, F32, F33]])
B = F.dot(F.T)
mpmath.sqrtm(Matrix(B))
然而,它给了我错误:
TypeError Traceback (most recent call last)
<ipython-input-14-439fed475a57> in <module>()
5 F = np.array([[F11, F12, F13], [F21, F22, F23], [F31, F32, F33]])
6 B = F.dot(F.T)
----> 7 mpmath.sqrtm(Matrix(B))
X:\WinPython3\python-3.4.2.amd64\lib\site-packages\sympy\mpmath\matrices\calculus.py in sqrtm(ctx, A, _may_rotate)
308
309 """
--> 310 A = ctx.matrix(A)
311 # Trivial
312 if A*0 == A:
X:\WinPython3\python-3.4.2.amd64\lib\site-packages\sympy\mpmath\matrices\matrices.py in __init__(self, *args, **kwargs)
326 A[i,j] = convert(A[i,j])
327 elif hasattr(args[0], 'tolist'):
--> 328 A = self.ctx.matrix(args[0].tolist())
329 self.__data = A._matrix__data
330 self.__rows = A._matrix__rows
X:\WinPython3\python-3.4.2.amd64\lib\site-packages\sympy\mpmath\matrices\matrices.py in __init__(self, *args, **kwargs)
299 for i, row in enumerate(A):
300 for j, a in enumerate(row):
--> 301 self[i, j] = convert(a)
302 else:
303 # interpret list as row vector
X:\WinPython3\python-3.4.2.amd64\lib\site-packages\sympy\mpmath\ctx_mp_python.py in convert(ctx, x, strings)
660 if hasattr(x, '_mpmath_'):
661 return ctx.convert(x._mpmath_(prec, rounding))
--> 662 return ctx._convert_fallback(x, strings)
663
664 def isnan(ctx, x):
X:\WinPython3\python-3.4.2.amd64\lib\site-packages\sympy\mpmath\ctx_mp.py in _convert_fallback(ctx, x, strings)
612 else:
613 raise ValueError("can only create mpf from zero-width interval")
--> 614 raise TypeError("cannot create mpf from " + repr(x))
615
616 def mpmathify(ctx, *args, **kwargs):
TypeError: cannot create mpf from F11**2 + F12**2 + F13**2
请问为什么会这样?这是sympy
的限制还是我做错了什么?
谢谢!
肖恩
答案 0 :(得分:1)
mpmath.sqrtm期待一个数字的矩阵;如果你想用B 中的每个元素的sqrt符号尝试:
>>> B.applyfunc(sqrt)
答案 1 :(得分:0)
不要使用NumPy进行符号计算。 NumPy仅适用于数值数组。
要获取矩阵的平方根,请使用B**(Rational(1, 2))
(sqrt(B)
也应该起作用,但默认情况下看起来仍未评估。)
在这种情况下,SymPy挂起,因为它通过对角化计算平方根,并且特征值不简化(或者至少SymPy不知道如何简化它们),因此它们是巨大的三次方程。看看B.eigenvals()
。因此,这个矩阵的平方根非常大。你期望平方根矩阵是一个相对简单的表达式吗?