scqubits.utils.plotting.wavefunction1d

scqubits.utils.plotting.wavefunction1d(wavefuncs, potential_vals, offset=0, scaling=None, **kwargs)[source]

Plots the amplitude of a single real-valued 1d wave function, along with the potential energy if provided.

Parameters:

  • wavefuncs (Union[WaveFunction, List[WaveFunction]]) – basis and amplitude data of wave function to be plotted

  • potential_vals (ndarray) – potential energies, array length must match basis array of wavefunc

  • offset (Union[float, Iterable[float]]) – y-offset for the wave function (e.g., shift by eigenenergy)

  • scaling (Optional[float]) – scaling factor for wave function amplitudes

  • **kwargs – standard plotting option (see separate documentation)

Return type:

Tuple[Figure, Axes]

Returns:

matplotlib objects for further editing