hw06.py 527 Bytes
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+1:n] = A [i,k+1:n] - lam*A[k, k+1:n]
            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 i in range(m):
	A.append(list(map(int, input().split(','))))
print(n,_)
b = list(map(int, input().split(',')))
Ab2Uc(A,b)