Tridiagonal matrix algorithm python

Print it