利用python实现杜利特尔分解法

杜利特尔分解法
博主目前大二在读，代码水平有限轻喷
使用的是python语言，适于数学者，通篇没有用到高级函数

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):
    U[0][i]=a[0][i]
for i in range(1,jieshu):
    L[i][0]=a[i][0]/U[0][0]
for i in range(2,jieshu+1):
    
    for j in range(i,jieshu+1):
        he=0
        for k in range(1,j):
            he+=L[i-1][k-1]*U[k-1][j-1]
        U[i-1][j-1]=a[i-1][j-1]-he
    for j in range(i+1,jieshu+1):
        he=0
        for k in range(1,j):
            he+=L[j-1][k-1]*U[k-1][i-1]
        L[j-1][i-1]=(a[j-1][i-1]-he)/U[i-1][i-1]
for i in range(jieshu):
    L[i][i]=1
print('L=',L)
print('U=',U)
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]
    else: 
        he=0
        for j in range(1,i+1):
            he+=L[i][j-1]*Y[j-1][0]
        Y[i][0]=b[i][0]-he
for i in range(jieshu):
    he1=0
    wuhu=jieshu-i
    if wuhu == jieshu:
        X[jieshu-1][0]=Y[jieshu-1][0]/U[jieshu-1][jieshu-1]
    else:
        for j in range(wuhu,jieshu):
            he1+=U[wuhu-1][j]*X[j][0]
        X[wuhu-1][0]=(Y[wuhu-1][0]-he1)/U[wuhu-1][wuhu-1]
print('x=',X)

上图为博主的输入案例