<s> """<STR_LIT>""" <EOL> """<STR_LIT>""" <EOL> import numpy as np <EOL> def gauss ( A , B , n , tol = <NUM_LIT> ) : <EOL> L = np . tril ( A ) <EOL> U = A - L <EOL> L_inv = np . linalg . inv ( L ) <EOL> x = np . zeros_like ( B ) <EOL> for i in range ( n ) : <EOL> Ux = np . dot ( U , x ) <EOL> x_new = np . dot ( L_inv , B - Ux ) <EOL> if np . allclose ( x , x_new , tol ) : <EOL> break <EOL> x = x_new <EOL> return x <EOL> A = np . array ( [ [ <NUM_LIT> , - <NUM_LIT:1.> , <NUM_LIT> , <NUM_LIT:0.> ] , <EOL> [ - <NUM_LIT:1.> , <NUM_LIT> , - <NUM_LIT:1.> , <NUM_LIT> ] , <EOL> [ <NUM_LIT> , - <NUM_LIT:1.> , <NUM_LIT> , - <NUM_LIT:1.> ] , <EOL> [ <NUM_LIT:0.0> , <NUM_LIT> , - <NUM_LIT:1.> , <NUM_LIT> ] ] ) <EOL> B = np . array ( [ <NUM_LIT> , <NUM_LIT> , - <NUM_LIT> , <NUM_LIT> ] ) <EOL> n = <NUM_LIT:100> <EOL> x = gauss ( A , B , n ) <EOL> print "<STR_LIT>" , x </s>
