Claudio Bellei cbellei

cbellei / TDMAsolver.py

Last active September 11, 2023 16:26 — forked from ofan666/TDMAsolver.py

Tridiagonal Matrix Algorithm solver in Python

	import numpy as np

	## Tri Diagonal Matrix Algorithm(a.k.a Thomas algorithm) solver
	def TDMAsolver(a, b, c, d):
	'''
	TDMA solver, a b c d can be NumPy array type or Python list type.
	refer to http://en.wikipedia.org/wiki/Tridiagonal_matrix_algorithm
	and to http://www.cfd-online.com/Wiki/Tridiagonal_matrix_algorithm_-_TDMA_(Thomas_algorithm)
	'''
	nf = len(d) # number of equations