hw06.py 589 Bytes
Newer Older
Kittisak Maneewong's avatar
Kittisak Maneewong committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25
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):
            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] = b[i] - lam * b[k]
    U = A.copy()
    c = b.copy()
    print(U)
    print(c)

m,n,_=map(int,input().split('x'))
print(m,n)
A = []
for row in range(m):
    #2,3,1,4,7
    A.append(list(map(int,input().split(','))))
print(n,_)
b = list(map(int,input().split(',')))
Ab2Uc(A,b)