Gauss elimination python numpy

# Importing NumPy Library
import numpy as np
import sys

# Reading number of unknowns
n = int(input('Enter number of unknowns: '))

# Making numpy array of n x n+1 size and initializing 
# to zero for storing augmented matrix
a = np.zeros((n,n+1))

# Making numpy array of n size and initializing 
# to zero for storing solution vector
x = np.zeros(n)

# Reading augmented matrix coefficients
print('Enter Augmented Matrix Coefficients:')
for i in range(n):
    for j in range(n+1):
        a[i][j] = float(input( 'a['+str(i)+']['+ str(j)+']='))

# Applying Gauss Elimination
for i in range(n):
    if a[i][i] == 0.0:
        sys.exit('Divide by zero detected!')
        
    for j in range(i+1, n):
        ratio = a[j][i]/a[i][i]
        
        for k in range(n+1):
            a[j][k] = a[j][k] - ratio * a[i][k]

# Back Substitution
x[n-1] = a[n-1][n]/a[n-1][n-1]

for i in range(n-2,-1,-1):
    x[i] = a[i][n]
    
    for j in range(i+1,n):
        x[i] = x[i] - a[i][j]*x[j]
    
    x[i] = x[i]/a[i][i]

# Displaying solution
print('\nRequired solution is: ')
for i in range(n):
    print('X%d = %0.2f' %(i,x[i]), end = '\t')

Source: codesansar.com

Similar codes

0 vote156 views1 answer
Seaborn dot plot hue
Python
0 vote155 views1 answer
Save dictionary to file numpy
Python
2 votes171 views1 answer
Open firefox python
Python
0 vote207 views1 answer
Exception Type: AssertionError Exception Value: Expected a `Response`, `HttpResponse` or `HttpStreamingResponse` to be returned from the view, but received a `<class 'django.db.models.query.QuerySet'>`
Python Django
0 vote149 views2 answers
How to use information from env variables in python
Python
7 votes143 views2 answers
What does // mean in python
Python
1 vote158 views1 answer
Django group by date from datetime field
Python
0 vote141 views2 answers
Business logic code where located in django rest framework
Python Django
9 votes146 views2 answers
Install python linux ubuntu
Python
0 vote147 views1 answer
Convert unit dynamo revit
Python