how to plot binomial tree with arrays in python

Question

I have a two tridiagonal matrix that (with equal dimensions) and want to plot it like the binomial tree in the picture below in python. binomial tree

here is my code