Up

week06/reame.py

-def ge(A, b):
-    # 1. elimination
-    n = len(b)
-    for j in range(0, n-1): # pivot equation
-        #print('------------\nj = %d'%(j))
-        for i in range(j+1, n):
-            lam = A[i,j]/A[j,j]
-            A[i,j:n] = A[i, j:n] - lam*A[j, j:n]
-            b[i] = b[i] - lam*b[j]
-            #print(A)
-            #print(b)
-            #print()
-    # 2. back substitution
-    x = b.copy()
-    for k in range(n-1, -1, -1):
-        x[k] = (b[k] - np.dot(A[k,k+1:n], x[k+1:n]))/A[k,k]
-    
-    print(x)