import kwant, numpy
from math import cos, sin
import matplotlib.pyplot as plt

lat = kwant.lattice.square()