As Majorana polarization is a vector field, I am trying to plot the values using quiver by adding the two functions:
def plot_vector_field(sys, Maj_x, Maj_y):
xmin, ymin = min(s.tag for s in sys.sites)
xmax, ymax = max(s.tag for s in sys.sites)
x, y = np.meshgrid(np.arange(xmin, xmax+1), np.arange(ymin, ymax+1))
plt.quiver(x, y, Maj_x, Maj_y)
plt.show()
def majorana_polarization(sys, energy):
wf = kwant.wave_function(sys, energy)
psi = wf(0)
eup, edown, hup, hdown = psi[::2], psi[1::2], psi[3::2], - psi[2::2]
Maj = eup * hup + edown * hdown
print(Maj)
Maj_x = np.real(Maj)
Maj_y = np.imag(Maj)
plot_vector_field(sys, Maj_x, Maj_y)
But since I am dealing with a honeycomb lattice the meshgrid will not work. How could I generate that using the sys sites?