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)