import numpy as np
import scipy.linalg as la
n = 6
np.random.seed(0)
A = (
+ np.diag(np.random.randn(n))
+ np.roll(np.diag(np.random.randn(n)), 1)
+ np.roll(np.diag(np.random.randn(n)), -1)
)
np.set_printoptions(precision=2)
A
P, L, U = la.lu(A)
L
U