ndarray

Question

예를 들어

의 인덱싱 문제는, 매트릭스가있다 :.

import numpy as np 
A = np.array([[ 8. , -6. , 2. ], 
       [-0.5, 8. , -6. ], 
       [ 0.5, -0.5, 2. ]]) 

그것은 LU 분해 (두리틀의 분해) 결과이다 (A = [L \ U]) 나는 L을 얻고 싶은
및 A.
에서 U U는해야한다 :

U = np.array([[ 8., -6., 2.], 
       [ 0., 8., -6.], 
       [ 0., 0., 2.]]) 

L가 있어야한다 :

L = np.array([[ 1. , 0. , 0. ], 
       [-0.5, 1. , 0. ], 
       [ 0.5, -0.5, 1.]]) 

그때 알고 싶습니다. L과 U를 어떻게 얻는 지 알고 싶습니까?

Answer 1

색인 조작이 필요하지 않습니다. 그냥 tril, triu 및 identity 기능을 사용하여 :

import numpy as np 
A = np.array([[ 8. , -6. , 2. ], 
       [-0.5, 8. , -6. ], 
       [ 0.5, -0.5, 2. ]]) 

U = np.triu(A) 

#[[ 8. -6. 2.] 
# [-0. 8. -6.] 
# [ 0. -0. 2.]] 

L = np.tril(A, k=-1) + np.identity(3) 

#[[ 1. 0. 0. ] 
# [-0.5 1. 0. ] 
# [ 0.5 -0.5 1. ]] 

Answer 2

당신이 나에게 LU 분해처럼 http://en.wikipedia.org/wiki/LU_decomposition

>>> U_ = np.array([[ 8., -6., 2.], 
       [ 0., 8., -6.], 
       [ 0., 0., 2.]]) 
>>> L_ = np.array([[ 1. , 0. , 0. ], 
       [-0.5, 1. , 0. ], 
       [ 0.5, -0.5, 1.]]) 
>>> np.dot(L_, U_) 
array([[ 8., -6., 2.], 
     [ -4., 11., -7.], 
     [ 4., -7., 6.]]) 

LU 분해를 보이지 않는 원하는 것은 scipy.linalg에서 사용할 수

>>> A = np.array([[ 8. , -6. , 2. ], [-0.5, 8. , -6. ], [ 0.5, -0.5, 2. ]]) 
>>> import scipy.linalg as spla 
>>> P, L, U = spla.lu(A) 
>>> L 
array([[ 1.  , 0.  , 0.  ], 
     [-0.0625 , 1.  , 0.  ], 
     [ 0.0625 , -0.01639344, 1.  ]]) 
>>> U 
array([[ 8.  , -6.  , 2.  ], 
     [ 0.  , 7.625  , -5.875  ], 
     [ 0.  , 0.  , 1.77868852]]) 
>>> np.dot(L, U) 
array([[ 8. , -6. , 2. ], 
     [-0.5, 8. , -6. ], 
     [ 0.5, -0.5, 2. ]])