pybinding graphene 周期性边缘态可视化

import pybinding as pb
import numpy as np
import matplotlib.pyplot as plt
from math import sqrt, pi
from pybinding.repository import graphene

pb.pltutils.use_style()

model = pb.Model(
    graphene.monolayer_4atom(),
    pb.rectangle(x=2, y=2),
    pb.translational_symmetry(a1=1.2, a2=False)
)
model.plot()
plt.show()

solver = pb.solver.lapack(model)
solver.set_wave_vector(k=0)
ldos = solver.calc_spatial_ldos(energy=0, broadening=0.01)  
ldos.plot(site_radius=(0.05, 0.05))
pb.pltutils.colorbar(label="LDOS")
plt.show()

还是不如kwant直观。