import matplotlib.pyplot as plt
from tenpy.models import lattice
plt.figure(figsize=(5, 4))
ax = plt.gca()
lat = lattice.Kagome(4, 4, None, bc='periodic')
lat.plot_coupling(ax, linewidth=3.)
lat.plot_order(ax, linestyle=':')
lat.plot_sites(ax)
lat.plot_basis(ax, origin=-0.25*(lat.basis[0] + lat.basis[1]))
ax.set_aspect('equal')
ax.set_xlim(-1)
ax.set_ylim(-1)
plt.show()