我可以找到每个要求解的方程,但似乎无法在原始矩阵中更新每个计算的结果。
我试图在矩阵中设置项目集,将u变量替换为solve函数的实际结果。但随后出现以下错误: TypeError:无法将序列乘以'Symbol'类型的非整数
不知道该怎么办!
from sympy import *
import numpy as np
#input matrix
a = np.array([[1,3,5],\
[2,4,7],\
[1,1,0]])
d = len(a)
d1 = d-1
d2 = (d1*(d1+1))/2
d3 = (d*(d+1))/2
symbols_dict = dict(('L%d'%k, symbols('L%d'%k)) for k in range(d2))
locals().update(symbols_dict)
symbols_dict = dict(('U%d'%k, symbols('U%d'%k)) for k in range(d3))
locals().update(symbols_dict)
# L decomposition
b = np.array([[1,0,0],\
[L0,1,0],\
[L1,L2,1]])
# U decomposition
c = np.array([[U0,U1,U2],\
[0,U3,U4],\
[0,0,U5]])
t=[[[]for i in range(d)]for j in range(d)]
s=[[[]for i in range(d)]for j in range(d)]
for j in range(d):
for i in range(d):
m=0
for k in range(d):
m = m+(b[j,k]*c[k,i])
t[j][i] = m
s[j][i]= solve(t[j][i]-a[j,i])
c.itemset((j,i),s[j][i])
应该为我解出每个方程式提供数值结果。
答案 0 :(得分:0)
更正后的代码如下:
L0,L1,L2 = symbols('L0 L1 L2')
U0,U1,U2,U3,U4,U5 = symbols('U0 U1 U2 U3 U4 U5')
.
.
.
for j in range(d):
for i in range(d):
m=0
for k in range(d):
m = m+(b[j,k]*c[k,i])
s= solve(m-a[j,i])
if j>i:
b[j,i]=s
else:
c[j,i]=s