I am going through the tutorials and I wanted to combine two of the lessons.

2.4. Beyond transport: Band structure and closed systems

2.6. Superconductors: orbital vs. lattice degrees of freedom

​In 2.4, they plot the wavefunction using kwant.plotter.map

​In 2.6, the system uses tinyarrays

Can I combine these and plot the wavefunction using map?

Here is my code

import kwant

import numpy as np

import matplotlib.pyplot as plt

import tinyarray as ta

s_0 = ta.array([[1, 0], [0, 1]])

s_x = ta.array([[0, 1], [1, 0]])

s_y = ta.array([[0, -1j], [1j, 0]])

s_z = ta.array([[1, 0], [0, -1]])

def make_system(a=1, L=21, W=21, t=1.0, mu=-3.618, Delta=0.1, leads=True):

    lat = kwant.lattice.square(a)

    syst = kwant.Builder()

    syst[(lat(x,y) for x in range(L) for y in range(W))] = -mu * s_z + Delta * s_x

    syst[lat.neighbors()] = -t * s_z

    if (leads):

        lead = kwant.Builder(kwant.TranslationalSymmetry((-a, 0)))

        lead[(lat(0, j) for j in range(W))] = -mu * s_z

        lead[lat.neighbors()] = -t * s_z

        syst.attach_lead(lead)

        syst.attach_lead(lead.reversed())

    return syst

def plot_conductance(syst, energies):

    data = []

    for energy in energies:

        smatrix = kwant.smatrix(syst, energy)

        # Conductance is N - R_ee + R_he

        data.append(smatrix.submatrix(0, 0).shape[0] -

                    smatrix.transmission(0, 0) +

                    smatrix.transmission(1, 0))

    plt.figure()

    plt.plot(energies, data)

    plt.xlabel("energy [t]")

    plt.ylabel("conductance [e^2/h]")

    plt.show()

def plot_wave_function(sys):

    # Calculate the wave functions in the system.

    ham_mat = sys.hamiltonian_submatrix(sparse=True)

    evecs = sla.eigsh(ham_mat, k=20, which='SM')[1]

    # Plot the probability density of the 10th eigenmode.

    kwant.plotter.map(sys, np.abs(evecs[:, 9])**2,

                      colorbar=False, oversampling=1)

syst = make_system()

kwant.plot(syst)

syst = syst.finalized()

plot_conductance(syst, energies=[-0.25 + 0.005 * i for i in range(100)])

plot_wave_function(syst)

Currently, I get the error "The number of sites doesn't match the number of provided values."

​Thanks,

Eric​