shnitsel.core.plot.spectra3d

Functions

spectra_all_times(inter_state)
inlabel(s, ax[, ha, va])
ski_plots(spectra)	Plot spectra for different times on top of each other,
pcm_plots(spectra)	Represent fosc as colour in a plot of fosc against time and energy.

Module Contents

spectra_all_times(inter_state)

Parameters:

inter_state (xarray.Dataset)

inlabel(s, ax, ha='center', va='center')

ski_plots(spectra)

Plot spectra for different times on top of each other, along with a dashed line that tracks the maximum. One plot per statecomb; plots stacked vertically. Expected to be used on data produced by spectra3d.spectra_all_times.

Parameters:

spectra (xarray.DataArray) – DataArray containing fosc values organized along ‘energy’, ‘time’ and ‘statecomb’ dimensions.

Return type:

Figure object corresponding to plot.

Examples

>>> from shnitsel.core import xrhelpers as xh, postprocess as P
>>> from shnitsel.core.plot import spectra3d
>>> spectra_data = (
        xh.open_frames(path)
        .pipe(P.get_inter_state)
        .pipe(P.assign_fosc)
        .pipe(spectra3d.spectra_all_times))
>>> spectra3d.ski_plots(spectra_data)

pcm_plots(spectra)

Represent fosc as colour in a plot of fosc against time and energy. The colour scale is logarithmic. One plot per statecomb; plots stacked horizontally. Expected to be used on data produced by spectra3d.spectra_all_times.

Parameters:

spectra (xarray.DataArray) – DataArray containing fosc values organized along ‘energy’, ‘time’ and ‘statecomb’ dimensions.

Return type:

Figure object corresponding to plot.

Examples

>>> from shnitsel.core import xrhelpers as xh, postprocess as P
>>> from shnitsel.core.plot import spectra3d
>>> spectra_data = (
        xh.open_frames(path)
        .pipe(P.get_inter_state)
        .pipe(P.assign_fosc)
        .pipe(spectra3d.spectra_all_times))
>>> spectra3d.pcm_plots(spectra_data)