Wavefunctions

For a qubit as simple as the transmon, wavefunctions are one-dimensional and can be plotted easily. The first option, is to employ the discrete charge basis (n):

transmon.plot_n_wavefunction(esys=None, which=0, mode='real');

Here, esys=None signifies that the eigensystem is not provided as an argument, but should be calculated fresh. which=0 specifies that the ground state wavefunction is to be plotted. Finally, mode='real' is asking for a plot of the real parts of the wavefunction amplitudes.

The options for mode are:

mode keyword	extracted function of amplitudes
'real'	real part of wavefunction amplitudes
'imag'	imaginary part of wavefunction amplitudes
'abs'	absolute value of wavefunction amplitudes
'abs_sqr'	absolute value squared of wavefunction amplitudes

Instead of the charge-basis representation, the transmon wavefunction can also be plotted in phi representation by switching from n to phi. Multiple wavefunctions can be plotted simultaneously by feeding a tuple to the argument which:

transmon.plot_phi_wavefunction(esys=None, which=[0,1,2,3,4], mode='real');