Metodo_Gauss_Seidel.py

# -*- coding: utf-8 -*-
"""
Spyder Editor

This is a temporary script file.
"""
#%%
import numpy
m=int(input('Valor de m:'))
n=int(input('Valor de n:'))
matrix = numpy.zeros((m,n))
x=numpy.zeros((m))

vector=numpy.zeros((n))
comp=numpy.zeros((m))
error=[]
print ('Método de Gauss-Seidel')
print ('Introduce la matriz de coeficientes y el vector solución')
for r in range(0,m):
    for c in range(0,n):
        matrix[(r),(c)]=float(input("Elemento a["+str(r+1)+str(c+1)+"] "))
    vector[(r)]=float(input('b['+str(r+1)+']: '))
print ("Método de Gauss-Seidel")


tol=float(input("Cual es la tolerancia que deseas? "))
itera=int(input("Cual es el numero max de iteraciones? "))
#Metodo de Gauss-Seidel

k=0
while k<itera: #Agregar
    suma=0
    k=k+1
    for r in range(0,m):
        suma=0
        for c in range(0,n):
            if (c != r):
                suma=suma+matrix[r,c]*x[c]               
        x[r]=(vector[r]-suma)/matrix[r,r]
        print("X["+str(r)+"]: "+str(x[r]))
    del error[:]
    #Comprobación
    for r in range(0,m):
        suma=0
        for c in range(0,n): 
            suma=suma+matrix[r,c]*x[c]    
        comp[r]=suma
        dif=abs(comp[r]-vector[r])
        error.append(dif)
        print("Error en x[",r,"]",error[r])
    print("Iteraciones: ",k)
    if all( i<=tol for i in error) == True:
        break
    
    print("Fin de la operacion")
#%%