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直观。