LU分解 python实现

最新推荐文章于 2023-09-21 20:43:13 发布

xckkcxxck

最新推荐文章于 2023-09-21 20:43:13 发布

阅读量7.4k

点赞数 2

分类专栏： python语法数值分析文章标签： python

本文链接：https://blog.csdn.net/xckkcxxck/article/details/78701515

版权

python语法同时被 2 个专栏收录

22 篇文章 3 订阅

订阅专栏

数值分析

6 篇文章 0 订阅

订阅专栏

以下是使用python实现LU分解，代码写的很丑，不要喷。。。

# -*- coding: utf-8 -*-
"""
Created on Sun Dec  3 10:03:45 2017

@author: www
"""

#   python实现LU分解

import numpy as np


def my_LU(B):
    A = np.array(B)
    n = len(A)
    #print(A)
    
    L = np.zeros(shape=(n,n))
    U = np.zeros(shape=(n,n))
    

    for k in range(n-1):
        gauss_vector = A[:,k]
        gauss_vector[k+1:] = gauss_vector[k+1:] / gauss_vector[k]
        gauss_vector[0:k+1] = np.zeros(k+1)
        #print(gauss_vector) 
        L[:,k] = gauss_vector
        L[k][k] = 1.0
        #print(L)
        #print(A)
        for l in range(k+1,n):
            B[l,:] = B[l,:] - gauss_vector[l] * B[k,:]
        
        A = np.array(B)
    L[k+1][k+1] = 1.0
    U = A
    print(U)
    print(L)
    
def main():
    A = np.array([[2., 2., 3.],
                  [4., 7., 7.],
                  [-2.,4., 5.]])
    my_LU(A)
        
     
main()