Create a band matrix class in python