gauss-elimination.py

import numpy as np
import sys

def gauss(a):
	n = len(a)
	m = len(a[0])

	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(m):
				a[j][k] = a[j][k] - ratio * a[i][k]

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

	for i in range(n):
		for j in range(m):
			a[i][j] = a[i][j] / a[i][i]

	return a


size = (3,5)
a = np.random.randint(100, size=(3,5))

print(a)
print(gauss(a))