Fix: Correct jacobi, gauss, cholesky
This commit is contained in:
22
ex1.py
22
ex1.py
@@ -1,22 +1,34 @@
|
||||
import numpy as np
|
||||
|
||||
|
||||
def resolve_up(A: np.array, b: np.array) -> np.array:
|
||||
def resolve_up(A, b) -> list[float]:
|
||||
"""
|
||||
|
||||
:param A: Une matrice
|
||||
:param b: Un vecteur
|
||||
:return:
|
||||
"""
|
||||
x = np.zeros(len(b))
|
||||
for i in range(len(A) - 1, -1, -1):
|
||||
right_of_diagonal = 0.0
|
||||
for k in range(i+1, len(A)):
|
||||
for k in range(i + 1, len(A)):
|
||||
right_of_diagonal += A[i, k] * x[k]
|
||||
x[i] = (b[i] - right_of_diagonal) / A[i,i]
|
||||
x[i] = (b[i] - right_of_diagonal) / A[i, i]
|
||||
return x
|
||||
|
||||
|
||||
def resolve_down(A: np.array, b: np.array) -> np.array:
|
||||
def resolve_down(A, b) -> list[float]:
|
||||
"""
|
||||
|
||||
:param A: Une matrice
|
||||
:param b: Un vecteur
|
||||
:return:
|
||||
"""
|
||||
x = np.zeros(len(b))
|
||||
for i in range(len(A)):
|
||||
sum = 0.0
|
||||
for k in range(i):
|
||||
sum += A[i,k] * x[k]
|
||||
sum += A[i, k] * x[k]
|
||||
x[i] = (b[i] - sum) / A[i, i]
|
||||
return x
|
||||
|
||||
|
||||
Reference in New Issue
Block a user