push week06/hw6.py

week06/hw06.py

+def Ab2Uc(A, b):
+    import numpy as np
+    A =np.array(A,float)
+    b =np.array(b,float)
+    n=len(b)
+    for k in range(0, n-1):
+        for i in range(k+1, n):
+            if A[i,k] != 0.0:
+                lam = A[i,k]/A[k,k]
+                A[i,k] = 0
+                A[i,k+1:n] = A[i, k+1:n] - lam*A[k,k+1:n]
+                b[i] -= lam*b[k]
+    print("U = ",A)
+    print("c = ",b)