利用python 解对称正定矩阵方程组的平方根法

大二在读，代码水平有限，轻喷

import math
a=[]
L=[]
U=[]
ling=0
jieshu=int(input("矩阵的阶数："))
print("请输入方程的系数矩阵")
for i in range(jieshu):
    onerow=[]
    Lonerow=[]
    Uonerow=[]
    for j in range(jieshu):
        x=float(input())
        onerow.append(x)
        Lonerow.append(ling)
        Uonerow.append(ling)
    a.append(onerow)
    L.append(Lonerow)
    U.append(Uonerow)
print("请输入方程等式右边的常数项")
b=[]
for i in range(jieshu):
    onerow1=[]
    for i in range(1):
        xx=float(input())
        onerow1.append(xx)
    b.append(onerow1)
for i in range(jieshu):
    if i == 0:
        L[0][0]=math.sqrt(a[0][0])
    else:
        L[i][0]=a[i][0]/L[0][0]
for j in range(1,jieshu):
    he=0
    for k in range(j):
        he+=(L[j][k])**2
    L[j][j]=math.sqrt(a[j][j]-he)
    for i in range(j,jieshu):
        he1=0
        for o in range(i):
            he1+=L[i][o]*L[j][o]
        L[i][j]=(a[i][j]-he1)/L[j][j]
print('L=',L)
print()
Y=[]
X=[]
for i in range(jieshu):
    onerow2=[]
    onerow3=[]
    for i in range(1):
        onerow2.append(0)
        onerow3.append(0)
    X.append(onerow2)
    Y.append(onerow3)
for i in range(jieshu):
    if i == 0:
        Y[0][0]=b[0][0]/L[0][0]
    else:
        he=0
        for k in range(i):
            he+=L[i][k]*Y[k][0]
        Y[i][0]=(b[i][0]-he)/L[i][i]
print('Y=',Y)
print()
for i in range(jieshu):
    wuhu=jieshu-i
    if wuhu==jieshu :
        X[wuhu-1][0]=Y[jieshu-1][0]/L[jieshu-1][jieshu-1]
    else:
        he=0
        for j in range(wuhu-1,jieshu):
            he+=L[j][wuhu-1]*X[j][0]
        X[wuhu-1][0]=(Y[wuhu-1][0]-he)/L[wuhu-1][wuhu-1]
print("X=",X)
        

输入案例