• Aruchira's avatar
    Up · 3a9a0753
    Aruchira authored
    3a9a0753
Name
Last commit
Last update
..
hw06.py Up
readme.md Up

def ge(A, b): # 1. elimination n = len(b) for j in range(0, n-1): # pivot equation

    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]
        
x = b.copy()
for k in range(n-1, -1, -1):
    x[k] = (b[k] - np.dot(A[k,k+1:n], x[k+1:n]))/A[k,k]

print(x)