import matplotlib.pyplot as plt
from tenpy.models import lattice
fig, axes = plt.subplots(1, 3, sharex=True, sharey=True, figsize=(8, 5))
lat = lattice.Square(5, 4, None, bc='periodic')
order_names=['default', 'snake', 'Fstyle']
for order_name, ax in zip(order_names, axes.flatten()):
    lat.plot_coupling(ax, linestyle='-', linewidth=1)
    lat.order = lat.ordering(order_name)
    lat.plot_order(ax, linestyle=':', linewidth=2)
    lat.plot_sites(ax)
    lat.plot_basis(ax, origin=(-0.5, -0.5))
    ax.set_title(f"order={order_name!r}")
    ax.set_xlim(-1, 5)
    ax.set_ylim(-1, 4)
    ax.set_aspect('equal')
plt.show()