hw06.py

def Ab2Uc(A,b):
    import numpy
    A = numpy.array(A,float)
    b = numpy.array(b,float)
    n = len(b)
    for j in range(0, n-1):
        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('U =',A)
        print('c =',b)